My Python Cheatsheet🔎

코딩테스트 준비를 하면서 파이썬을 파이썬답게 활용하기 위한 문법을 정리한 문서이다.
공책에 정리한 것들을 옮겨가면서 예제를 만들고 있어 시간이 될 때마다 추가할 예정이다.
화이팅,,나 자신,,

[2022.02.20] 최초 작성
[2022.02.21] 내용 추가
[2022.02.27] 내용 추가
[2022.03.29] 내용 추가

🎈a//b, a%b 한번에 출력

  a = 11
  b = 5

  print(*divmod(a,b)) # 2 1

* : 패킹 혹은 언패킹 기호
인자 앞에서 튜플, 리스트 등을 풀어주는 역할(unpacking)을 한다.

🎈n진법을 10진수로 변경, 문자열을 숫자로 변경

  base = 2
  num = "1011" 
  print(int(num, base))  # 11

공식 문서 : int( )
base진법의 숫자num을 십진수로 바꿔서 출력해준다.
이때 num의 형식은 string이다. base는 생략이 가능하고, 생략시 string을 int로 바꿔주는 역할을 한다.
보통 문자열을 숫자로 바꿀 때 많이 사용했던 것 같다. 반대(숫자->문자) 는 str() 을 사용한다.

🎈문자열 배치하기

	string1 = "hello"
    n = 10

    print(string1.ljust(n))  # hello     
    print(string1.center(n)) #   hello   
    print(string1.rjust(n))  #      hello

string1.ljust(n) : n칸만큼의 공백을 두고 string1 좌측정렬
tring1.center(n) : n/2칸 사이에서 string1 중앙정렬
string1.rjust(n) : n칸만큼의 공백을 두고 string1 우측정렬

🎈string 상수 활용하기

	import string

    print(string.ascii_lowercase)  # abcdefghijklmnopqrstuvwxyz 
    print(string.ascii_uppercase) # ABCDEFGHIJKLMNOPQRSTUVWXYZ
    print(string.ascii_letters)  # abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ
    print(string.digits)  # 0123456789
    

말 그대로 상수, 고정돼있는 값이다. 하나하나 적을 필요 없이 불러올 수 있다는 점이 편리하다.

🎈리스트 정렬하기

기본 정렬하기

	testList = [3,5,7,4,8,5,8,10]
    
    print(testList)  # [3, 5, 7, 4, 8, 5, 8, 10]
    testList.sort() # 원본 직접 변경
    print(sorted(testList))  # 원본 유지, 보통 새로운 리스트로 할당해서 사용
    
    # 결과는 [3, 4, 5, 5, 7, 8, 8, 10]로 동일
    

공식문서 : ❓ sort( ) ❓ sorted( )
보통 원본 리스트를 보존해야 하는 경우는 sorted(listname)를 사용하고, 그냥 바꿀 때에는 listname.sort()를 사용한다.
근데, sort하려는 리스트의 자료형이 문자형 일땐 어떻게 해야할까??

리스트의 요소가 문자형일때 정렬하기

	testList1 = ['1', '10', '2', '200', '22', '23', '3', '4']
    testList2 = ['1', '10', '2', '200', '22', '23', '3', '4']
    
    print(testList1)  # ['1', '10', '2', '200', '22', '23', '3', '4']
    print(testList2)  # ['1', '10', '2', '200', '22', '23', '3', '4']
    testList1.sort() # 문자형 순자의 문자 순서대로 정렬
    testList2.sort(key = int) # 문자형 숫자를 숫자로 취급하여 정렬
    print()
    print(testList1)  # ['1', '10', '2', '200', '22', '23', '3', '4']
    print(testList2)  # ['1', '2', '3', '4', '10', '22', '23', '200']
    

문자형으로 이루어진 숫자는 그냥 정렬했을 경우에는 testList1 처럼 숫자를 문자처럼 생각하여 순서대로 정렬한다.
testList2는 key=int라는 인자를 통해서, 하나하나 int자료형으로 바꾸지 않더라도 숫자처럼 생각하여 정렬한다. key 인자는 lambda를 사용하는 것도 가능하다. key=int 를 사용하지 않고, 하나하나 바꾸는 것은 map()을 이용해 구현가능하다.

리스트 뒤집기

	testList1 = [10,5,7,8,3]
    testList2 = [10,5,7,8,3]
    
    
    print(testList1)  # [10, 5, 7, 8, 3]
    print(testList2)  # [10, 5, 7, 8, 3]
    testList1.reverse() # 단순 뒤집기
    testList2.sort(reverse = True) # 역순으로 정렬하기(내림차순)
    print()
    print(testList1)  # [3, 8, 7, 5, 10]
    print(testList2)  # [10, 8, 7, 5, 3]

listName.reverse()는 말 그대로 기존 리스트를 뒤집어서 배열한다.
listName.sort(reverse = True)는 정렬을 역순으로 할 수 있도록 한다. 하지만 key=int랑 같이 쓰는건 안된다. 왜 그런지는 모름

🎈map() 활용하기

🧡 최애함수

	testList = ["10","5","7","8","3"]
    testList2 = [("i","love","you"),("you","love","me")]
    
    map_list = list(map(int,testList))
    print(map_list)  # [10, 5, 7, 8, 3]
    
    map_list2 = list(map(''.join,testList2))
    print(map_list2) #['iloveyou', 'youloveme']

공식문서 : ❓ map( )
list(map(func,data))형식으로 많이 쓴다. data에는 순환할 수 있는 값인 list, tuple, dict등이 들어간다.
func에는 각 값에 적용될 함수 이름을 쓴다. ()쓰면 안된다
list를 안 씌우면 map 객체라는 것으로 반환이 되는데, 이거는 어디에 뭐 쓸모도 없는 객체라 꼭 list(), tuple()등 내장함수에 담아야 한다.

🎈 zip(), 두 그룹의 데이터를 서로 엮어주는 방법

zip() 사용하기

	a = [1,2,3]
    b = ['a','b','c']
    
    print(list(zip(a,b))) # [(1, 'a'), (2, 'b'), (3, 'c')]
    print(dict(zip(a,b))) # {1: 'a', 2: 'b', 3: 'c'}

공식문서 : ❓ zip( )
그냥 사용할 경우 map()처럼 zip객체 형식으로 반환되기 때문에 list(), dict()등으로 묶어서 사용해야 한다.
특히 두 리스트를 사전형식으로 합칠 때 for문을 사용하지 않고도 간편하게 해결가능하다.
두 리스트의 길이가 다르다면, 짧은 리스트 기준으로 zip 후 남은 리스트는 버린다.

zip() 해체하기

	a = [1,2,3]
    b = ['a','b','c']

    zip_a_and_b = list(zip(a,b)) # [(1, 'a'), (2, 'b'), (3, 'c')]

    unzip_a, unzip_b = list(zip(*zip_a_and_b))
    print(unzip_a, unzip_b)  # (1, 2, 3) ('a', 'b', 'c')

언패킹 기호인 * 을 활용해서 unzip이 가능하다.

그러면 하나의 리스트를 zip형식으로 묶는 방법 은 무엇일까???

리스트 한 개로 zip() 구현하기

    a = [1,2,3,4,5,6]

    for i, j in zip(a,a[1:]) :
        print(i, j)

결과

	1 2
    2 3
    3 4
    4 5
    5 6

i와 j에 각각 1~5와 2~6까지 각각 할당이 가능한 모습이다. 아직까지 써 본 적은 없지만..?..

🎈 리스트의 원소가 문자일때, 문자열로 이어붙이기

    a = ['I','hate','paprika']

    a_join = ''.join(a)
    print(a_join) # Ihatepaprika

정말정말정말 많이 쓴다.
list의 원소를 하나로 이어붙인다. 이중 ''는 sep문자이고 원소가 문자 형식일때만 사용 가능하다.

🎈 문자열을 쉽게 반복하기

	a = [456,789]
    print('abc'*10) # abcabcabcabcabcabcabcabcabcabc
    print(a*5) # [456, 789, 456, 789, 456, 789, 456, 789, 456, 789]

* 문자는 기능이 참 많다,, 이거는 위에 패킹/언패킹 기능보다는 연산자로 활용하는 방법이다.
일반 문자열에 사용하면 곱하기 n회 기능으로, 문자열을 n번 반복한다.
리스트에 넣으면 내부 값이 변하는게 아니라 리스트의 원소가 반복된다.

🎈 2차원 리스트를 1차원 리스트로 바꾸기

• 시간은 numpy < itertools < list comprension < sum() 순으로 짧다. 시간이 짧을수록 좋은 코드이다.

sum() 함수 사용하기

	mylist = [[1,2,3], [100,200,300], [10,20,30]]
    print(sum(mylist, [])) # [1, 2, 3, 100, 200, 300, 10, 20, 30]

공식 문서 : ❓ sum( )
sum()을 통해 한겹씩 벗겨내는 방법이다. 간단하지만 속도가 제일 느려 추천하지 않는 방법이다.

chain 을 이용하는 방법

	import itertools
    
    mylist = [[1,2,3], [100,200,300], [10,20,30]]
    print(list(itertools.chain.from_iterable(mylist))) # [1, 2, 3, 100, 200, 300, 10, 20, 30]
    print(list(itertools.chain(*mylist))) # [1, 2, 3, 100, 200, 300, 10, 20, 30]
    

공식문서 : ❓ itertools ❓ chain()
itertools에서 제공하는 chain() 을 이용하는 방법이다. chain()은 기본적으로 인자로 리스트나 튜플 등을 여러개 주면 그걸 하나로 이어준다. 하지만 다차원 리스트를 연결하려면, 인자의 리스트를 *을 통해 언패킹 해서 집어넣거나, 언패킹 하지 않고 chain.from_iterable() 를 이용하면 된다.

reduce() 를 이용하는 방법 1

	from functools import reduce
    
    mylist = [[1,2,3], [100,200,300], [10,20,30]]
    print(list(reduce(lambda x,y : x+y, mylist))) # [1, 2, 3, 100, 200, 300, 10, 20, 30]

공식문서 : ❓ functools ❓reduce( )
reduce() 안에 람다식을 사용하여 구현하였다.

reduce() 를 이용하는 방법 2

	from functools import reduce
	import operator

	mylist = [[1,2,3], [100,200,300], [10,20,30]]
	print(list(reduce(operator.add, mylist))) # [1, 2, 3, 100, 200, 300, 10, 20, 30]

공식문서 : ❓ operator

아까와 동일하게 reduce()를 이용하지만 operator.add라는 덧셈 연산자를 통해 새로운 리스트에 하나씩 원소를 더해가는 방식이다.

List Comprension 을 이용하는 방법

	mylist = [[1,2,3], [100,200,300], [10,20,30]]
    print([element for arr in mylist for element in arr]) # [1, 2, 3, 100, 200, 300, 10, 20, 30]

꽤 간단하다. 하지만 개인적으로 항상 헷갈리는 나를 위해 정리해두자..ㅎㅎ

이걸 일반적인 for문으로 풀이해보면,

	mylist = [[1,2,3], [100,200,300], [10,20,30]]
    element = []
    mylist2 = []
    for arr in mylist:
        for element in arr:
            print("element : ", element)
            mylist2.append(element)
    
    print(mylist2) # [1, 2, 3, 100, 200, 300, 10, 20, 30]

결과 :

	element :  1
    element :  2
    element :  3
    element :  100
    element :  200
    element :  300
    element :  10
    element :  20
    element :  30
    [1, 2, 3, 100, 200, 300, 10, 20, 30]

리스트 컴프리헨션을 통해 이중for문을 한줄로 나타낼 수 있다는 점이 매력적이다.

numpy 를 이용하는 방법

	import numpy as np
    
    mylist = [[1,2,3], [100,200,300], [10,20,30]]
    print(np.array(mylist).flatten().tolist()) # [1, 2, 3, 100, 200, 300, 10, 20, 30]

공식 문서 : ❓ numpy ❓ array( ) ❓ flatten( )

이 방법은 [ [A리스트], [B리스트] ] 에서 A리스트와 B리스트의 원소의 길이가 같아야 가능하다.

🎈 순열과 조합

공식문서 : ❓ itertools

import itertools

list2 = ['1','2','3','4'] # 원소가 str 이어야 함
n = 3

# 모든 원소로 순열 만들기
print("순열",list(map(''.join, itertools.permutations(list2))))

# n개의 원소로 순열 만들기
print("N-순열",list(map(''.join, itertools.permutations(list2, n))))

# 조합 
print("조합",list(map(''.join, itertools.combinations(list2, n)))) 

결과 :

🎈 가장 많이 등장하는 알파벳 구하기

공식문서 : ❓ collections.Counter

import collections

list2 = 'aaaaaaaaaabcd'

counter = collections.Counter(list2)
print(counter.most_common()) # [('a', 10), ('b', 1), ('c', 1), ('d', 1)]

collections.Counter() 는 반환값이 counter-dict 객체이다. 이걸 most_common을 사용하면 , 사용한 알파벳만 리스트(튜플) 형태로 반환한다.

🎈 for...else 문

쉽게 말하면 for문이 break없이 잘 마무리 되었을 때 else문을 통과시킨다.

break 가 있을 때

list2 = [1,2,3,4]

for i in list2:
    if i >= 4 :
        break
    print(i)

else :
    print("for문이 잘 마무리 됨")

결과 :

break 가 없을 때 >> else문 통과

list2 = [1,2,3,4]

for i in list2:
    #if i >= 4 : 
    #   break
    print(i)

else :
    print("for문이 잘 마무리 됨")

결과 :

🎈 간단하게 두 가지 변수 swap 하기

a = 'yu'
b = 'jin'

print(a,b) # yu jin

a,b = b,a

print(a,b) # jin yu

파이썬은 정말,, 마법의 언어다.