파이썬을 파이썬 답게

오늘은 파이썬 공부할 때 많이 유용했던 프로그래머스의 '파이썬을 파이썬 답게' 강의에 대해 소개해드리고자 해요!

파이썬을 파이썬답게

 

파이썬 왠만한 문법에 대해 다 잘 알고 있다고 생각했는데 '파이썬을 파이썬답게' 강의와 문제를 풀어보면서 아직 많이 부족하고, 파이썬은 정말 무궁무진한 언어라는 것을 알았어요! 알고리즘 문제 풀 때 파이썬이 얼마나 좋은 언어인지도 다시한번 깨달았습니다 😀

 

그래서 ! '파이썬을 파이썬 답게' 강의에서 새롭게 배운 유용한 파이썬 문법/모듈에 대해 정리해볼게요!

'파이썬을 파이썬 답게' 강의 요약 및 정리라고 생각하시면 될 거 같습니다 : )

 

 

📌 몫과 나머지를 구할 때?! - divmod

보통을 몫과 나머지를 구할 때 어떻게 구하시나요!? 아래와 같이 연산자를 이용하여 몫, 나머지 따로 구하실 거 같은데요.

divmod를 이용하면 보다 쉽게 몫 과 나머지를 동시에 구하실 수 있습니다.

a = 7
b = 5
print( a//b , a%b ) # 1 2 출력

a = 7
b = 5
print( *divmod(a, b) ) # 1 2 출력

 

 

📌 n 진법 표기 된 string을 10 진법 숫자로 변환하기 - int

string으로 표기된 n 진법 숫자를 10 진법으로 변환하는 문제가 있었습니다.

제가 알고 있던 int는 Integer 형으로 변환을 해주는 역할만 한다고 생각했었는데 이번 기회에 10진법 변환도 자동으로 해준다는 사실을 배웠습니다!

파이썬을 사용하면 진법 변환도 매우 쉽게 할 수 있다는 걸 배우게 되었어요.

num = '3212'
base = 5

st = [int(i) for i in str(num)] # string 숫자를 한글자씩 쪼개어 int형으로 list에 저장하기
st.reverse() # 진법에 맞게 곱해주기 위하여 list 거꾸로 하기
answer = 0 # 10진법으로 변환된 숫자 더해주기

for i in range(len(st)):
  answer += st[i]*(base**i) 

num = '3212'
base = 5

answer = 0
for idx, i in enumerate(num[::-1]):
    answer += int(i) * ( base ** idx )

이렇게 꽤 긴 수식을 int를 통해 단 한번에 10진법으로 진법 변환이 가능합니다.

 

num = '3212'
base = 5

answer = int(num, base)

 

 

📌 문자열 정렬하기 - ljust, center, rjust

문자열에 대한 함수들을 정리해논 적 있었는데, ljust,center,rjust의 사용법에 대해서는 잘 모르고 있었습니다.

이번에 문자열 정렬 문제를 풀면서 이렇게 사용되는 구나 깨달았습니다.

 

📝 문자열 s를 좌측 / 가운데 / 우측 정렬한 길이 n인 문자열을 한 줄씩 프린트해보기

s = '가나다라'
n = 7

s.ljust(n) # 좌측 정렬
s.center(n) # 가운데 정렬
s.rjust(n) # 우측 정렬

 

 

📌 알파벳 출력하기 - string 모듈

보통 'a' 부터 'z'까지 출력하시오 하면 'abcd....xyz' 이렇게 하나하나 나열하는 경우가 많았을 거 같은데요.

저 또한 하나하나 불필요하게 나열했던 사람 중 한명이였습니다.

string 모듈에는 소문자, 대문자, 대소문자 모두, 숫자를 출력해주는 기능이 있었습니다. 

string 모듈에 대해 더 많이 공부해야 겠어요!

 

import string 

string.ascii_lowercase # 소문자 abcdefghijklmnopqrstuvwxyz
string.ascii_uppercase # 대문자 ABCDEFGHIJKLMNOPQRSTUVWXYZ
string.ascii_letters #대소문자 모두 abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ
string.digits # 숫자 0123456789

 

 

📌 2차원 리스트 뒤집기 - zip

' [ [1,2,3], [4,5,6], [7,8,9] ] 리스트가 주어진 경우, [[1, 4, 7], [2, 5, 8], [3, 6, 9]] 로 뒤집어라 ' 라는 문제를 보고 어떻게 뒤집을 수 있을까 생각해보다 리스트의 행과 열의 길이가 같다는 조건을 보고, for문을 통해 temp 리스트에 1행 1열, 2행 1열 .. 순차적으로 저장하여 정답 리스트에 더해주는 형식으로 구현해보았는데요.

 

zip을 이용하면 보다 수월하게 2차원 리스트를 뒤집을 수 있다는 것을 배웠습니다.

def solution(mylist):
  answer = []
  for i in range(len(mylist[0])):
    tmp = []
    for j in range(len(mylist)):
      tmp.append(mylist[j][i])
    answer.append(tmp)  
    
  return answer

mylist = [ [1,2,3], [4,5,6], [7,8,9] ]
new_list = list(map(list, zip(*mylist)))

 

 

추가로 zip의 유용한 사용 예시도 정리해보았습니다.

 

✔️여러 개의 Iterable 동시에 순회할 때 

list1 = [1, 2, 3, 4]
list2 = [100, 120, 30, 300]
list3 = [392, 2, 33, 1]
answer = []
for i, j, k in zip(list1, list2, list3):
   print( i + j + k )

 

✔️Key 리스트와 Value 리스트로 딕셔너리 생성할 때

animals = ['cat', 'dog', 'lion']
sounds = ['meow', 'woof', 'roar']
answer = dict(zip(animals, sounds)) # {'cat': 'meow', 'dog': 'woof', 'lion': 'roar'}

 

 

📌 리스트의 모든 원소의 type 변환하기 - map

map 은 input으로 여러개의 숫자를 한 줄로 받을 때 많이 사용하는 함수라 꽤 익숙 하실거 같은데요.

map을 응용한 방법에 대해도 소개 해드리고자 합니다.

list1 = ['1', '100', '33']
list2 = list(map(int, list1)) # [1, 100, 33]

 

✔️이차원 리스트의 원소의 길이를 담은 리스트로 만들때

[ [1,2] , [3,4,5], [1] ] 👉🏻 [ 2, 3, 1 ] 로 만들기

answer = list(map(len,mylist))

 

항상 map을 사용할 때 int 만 넣어본 거 같은데 내가 정의한 함수나, len과 같은 함수들을 넣어도 자동으로 인자들을 함수에 넣어 변환해주는 성질을 같고 있다는 것을 배울 수 있었습니다.

 

 

📌 sequence 원소들을 하나로 이어 붙이기 - join

• 문자열 배열 ['1', '100', '33']을 이어 붙여 문자열 '110033' 만들기
• 정수형 튜플 (3, 22, 91)을 이어붙여 문자열 '32291' 만들기

위와 같은 상황에서 어떻게 붙일 수 있을까 ?! 막막한 경우가 있으셨을 텐데 파이썬에서는 join을 통해 한번에 string으로 이어 붙이실 수 있습니다.

my_list = ['1', '100', '33']
answer = ''.join(my_list) # '110033'

 

✔️String을 List로 변환하려면 !?

str = '11 22 33'
list = str.split( ) # ['11', '22', '33']

 

 

📌곱집합(Cartesian product) 구하기 - product

product에 대해서는 알고리즘 문제 풀때 자주 보진 못한 것 같지만 알아두면 훨씬 빠르고 쉽게 사용할 수 있는 함수 중 하나인 것 같습니다!

 

두 스트링 'ABCD', 'xy' 의 곱집합인 Ax Ay Bx By Cx Cy Dx Dy 구할 때 / 2개이상의 iterable 곱집합 구할 때

import itertools

iterable1 = 'ABCD'
iterable2 = 'xy'
iterable3 = '1234'
itertools.product(iterable1, iterable2, iterable3) 

 

 

📌 2차원 리스트를 1차원 리스트로 만들기 - from_iterable

itertools를 이용하면 쉽게 2차원 리스트를 1차원 리스트로 만들 수 있습니다!

2차원 리스트를 1차원 리스트로 만드는 방법 중 간단한 방법에 대해서 소개해드릴게요.

 

import itertools
my_list = [[1, 2], [3, 4], [5, 6]]

# 방법 1 - sum 함수
answer = sum(my_list, [])

# 방법 2 - itertools.chainimport itertools
list(itertools.chain.from_iterable(my_list))

# 방법 3 - itertools와 unpackingimport itertools
list(itertools.chain(*my_list))

# 방법4 - list comprehension 이용
[element for array in my_list for element in array]

 

 

📌 순열과 조합 - combinations, permutations

iterable의 원소를 가지고 순열과 조합을 구하는 방법에 대해 알려드리고자 합니다.

순열과 조합을 구하는 방식은 원소의 개수가 적으면 쉽게 구하실 수 있지만 알고리즘을 구하기에는 어려운 문제이지 않을까 싶습니다.

itertools의 permutations을 이용하면 보다 쉽게 구하실 수 있습니다!

• 1,2,3의 숫자가 적힌 카드가 있을 때, 이 중 두 장을 꺼내는 경우의 수 -> 12, 13, 21, 23, 31, 32
• 'A', 'B', 'C'로 만들 수 있는 경우의 수 -> 'ABC', 'ACB', 'BAC', 'BCA', 'CAB', 'CBA'

import itertools

pool = ['A', 'B', 'C']
print(list(map(''.join, itertools.permutations(pool)))) # 3개의 원소로 수열 만들기
print(list(map(''.join, itertools.permutations(pool, 2)))) # 2개의 원소로 수열 만들기

 

 

📌 가장 많이 등장하는 알파벳 찾기 - Counter

collections 모듈에 존재하는 Counter는 꽤 많이 사용해보셨을 거라고 생각합니다.

하지만 ' mystr에서 가장 많이 등장하는 알파벳만을 사전 순으로 출력하기 ' 이런 문제를 받았을 때 어떻게 문제를 풀어야 할지 바로 생각나지 않았는데요. Counter을 이용하면 쉽게 풀 수 있다는 걸 배울 수 있었습니다.

• 'aab' -> 'a'
• 'dfdefdgf' -> 'df'
• 'bbaa' -> 'ab'

from collections import Counter
my_str = input().strip()

counter = Counter(my_str).most_common()
answer =""

for i in range(len(counter)):
if counter[i][1] == counter[0][1]:

answer+=counter[i][0]

print(''.join(sorted(list(answer))))

 

collections의 Counter 클래스에는 원소가 몇번 등장하는지 구하는 것 말고도 most_common() 처럼 데이터의 개수가 많은 순으로 정렬된 리스트를 리턴하는 메서드도 존재합니다.

from collections import Counter

my_list = [1, 2, 3, 4, 5, 6, 7, 8, 7, 9, 1, 2, 3, 3, 5, 2, 6, 8, 9, 0, 1, 1, 4, 7, 0]
answer = Counter(my_list)

print(answer[1]) # = 4
print(answer[3])  # = 3
print(answer[100]) # = 0

 

 

📌 for 문과 if문을 한번에 - List comprehension의 if 문

list 원소들 중 2의 배수만을 2배 하여 저장하고 싶다면 ?! 파이썬에서는 for문과 if문을 한번에 쓸 수 있는 List의 comprehension이 존재합니다.

mylist = [3, 2, 6, 7]
answer = [ i**2 for i in mylist if i %2 == 0] # [4, 12]

 

 

📌 flag OR for - else

for 문을 돌려 조건에 해당하는 원소가 존재하는지 확인한 다음에 존재하면 True를 , 존재하지 않으면 Fasle와 같이 출력해야하는 경우가 존재할 수 있습니다. 이때 for - else 문을 사용하면 간편하게 코드를 짜실 수 있습니다.

 

문제 )

1. 숫자를 차례로 곱해 나온 수가 제곱수1가 되면 found를 출력하고
2. 모든 수를 곱해도 제곱수가 나오지 않았다면 not found를 출력하기

import math

numbers = [int(input()) for _ in range(5)]
multiplied = 1
for number in numbers:
    multiplied *= number
    if math.sqrt(multiplied) == int(math.sqrt(multiplied)):
        print('found')
        break
else:
    print('not found')

 

 

📌 두 변수의 값 바꾸기 - swap

swap을 하기 위해서는 보통 temp를 이용하여 변환하는 방법에 대해 익숙하실 텐데요.

파이썬에서는 간단하게 swap를 하실 수 있습니다!

a = 3
b = 'abc'

a, b = b, a # a = 'abc', b = 3

 

 

📌이진 탐색 하기 (binary search) - bisect

이진 탐색이란 ?! 오름차순으로 정렬된 리스트에서 특정 값 위치를 찾아내는 알고리즘으로 검색속도가 빠르다는 장점이 있습니다.

알고리즘 문제를 푸실 때 이진 탐색을 사용하는 문제가 많이 등장하는데요. 이때 이진 탐색 알고리즘을 작성하는 것이 아닌 bisect모듈의 bisect 를 이용하시면 간단하게 이진 탐색을 이용하실 수 있습니다.

 

이진 탐색 알고리즘

def bisect(a, x, lo=0, hi=None):
    if lo < 0:
        raise ValueError('lo must be non-negative')
    if hi is None:
        hi = len(a)
    while lo < hi:
        mid = (lo + hi) // 2
        if a[mid] < x:
            lo = mid + 1
        else:
            hi = mid
    return lo

mylist = [1, 2, 3, 7, 9, 11, 33]
print(bisect(mylist, 3))

 

bisect 모듈을 이용한 이진 탐색

import bisect
mylist = [1, 2, 3, 7, 9, 11, 33]
print(bisect.bisect(mylist, 3))

 

 

📌클래스의 인스턴스 출력 - class의 자동 string casting

class를 생성한 뒤 print문을 추가로 출력하고 싶을 때가 있습니다.

이때 파이썬에서는 __str__ 메소드를 이용하여 class 내부에서 출력 format을 지정하는 것이 가능합니다!

class Coord(object):
    def __init__ (self, x, y):
        self.x, self.y = x, y

point = Coord(1, 2)
print( '({}, {})'.format(point.x, point.y) ) 

# 또는
def print_coord(coord):
    print( '({}, {})'.format(coord.x, coord.y) )
print_coord(point)

class Coord(object):
    def __init__ (self, x, y):
        self.x, self.y = x, y
    def __str__ (self):
        return '({}, {})'.format(self.x, self.y)

point = Coord(1, 2)

 

 

📌 가장 큰수 - inf

inf를 이용하면 아무리 큰 어떤 숫자와 비교하더라도 무조건 크다고 판정하실 수 있습니다. 

inf에는 음수 기호를 붙일 수 있다는 점도 기억해주시면 좋을거 같아요.

min_val = float('inf')
min_val > 10000000000

max_val = float('-inf')

 

 

📌 파일 입출력 간단하게 하기

파이썬의 with - as 구문을 이용하게 되면 파일을 close하지 않고 자동 close하실 수 있고, 

readlines시 파일의 EOF 까지만 읽어, while문을 통해 EOF를 체크할 필요가 없습니다.

with - as 구문은 파일 뿐만아니라 socket, http에서도 사용하실 수 있습니다!

 

with open('myfile.txt') as file:
  for line in file.readlines():
    print(line.strip().split('\t'))

 

 

 

이번 포스팅에서는 python의 유용한 문법과 모듈에 대해 알아보았는데요. 

저는 '파이썬을 파이썬 답게' 강의를 들으면서 아주 유익한 파이썬 코드 작성하는 법에 대해 배울 수 있었던 것 같습니다. 

파이썬으로 알고리즘 문제를 푸시는 분들이라면 매우 추천하는 강의 입니다! : )