[프로그래머스] '파이썬을 파이썬답게' 강의 정리

프로그래머스 '파이썬을 파이썬답게' 강의 정리

https://programmers.co.kr/learn/courses/4008

파이썬을 파이썬답게

1. 몫과 나머지 : divmod() 함수 사용

# 다른 언어
a = 7
b = 5
print(a//b, a%b) 

# 파이썬
a = 7
b = 5
print(*divmod(a,b))

 

2. n진법으로 표기된 string 을 10진법 숫자로 변환하기

# 파이썬
# base(5)진법으로 표기된 num(3212)를 10진법으로 변환하는 방법
# num 이 string 이 아닌 경우, num = str(num) 추가 필요
num = '3212' 
base = 5
print = int(num,base)

 

3. 문자열 정렬하기 : ljust, center, rjust

# 파이썬
n = 7  #7칸
s = '가나다라'

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

 

4. 알파벳 출력하기

# 파이썬
import string

string.ascii_lowercase	# 소문자 abcdefgh...
string.ascii_uppercase	# 대문자 ABCDEFGH...
string.ascii_letters	# 대소문자 모두 abcdef...xyzABCD...
string.digits			# 숫자 0123...9

# str type으로 특정 알파벳, 숫자를 뽑으려면 인덱싱하면됨

 

5. 원본 유지한채, 정렬된 리스트 구하기 : sorted()

# 파이썬
list1 = [3,2,1]
list2 = sorted(list1)

 

6. 2차원 리스트 뒤집기 : zip, unpacking 을 사용

* 전치 행렬을 구하는 방법이며, numpy 사용 못할 때 2중 for문 보다 간단하다

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

* zip(*iterables)은 각 iterables 의 요소들을 모으는 이터레이터를 만든다. 튜플의 이터레이터를 돌려주는데, i 번째 튜플은 각인자로 전달된 시퀀스나 이터러블의 i 번째 요소를 포함한다.

# zip 사용법
mylist = [1, 2, 3]
new_list = [40, 50, 60]
for i in zip(mylist, new_list):
    print (i)

(1, 40)
(2, 50)
(3, 60)

 

7. i 번째 원소와 i+1 번째 원소의 차를 담은 일차원 리스트에 차례로 담아 리턴하는 방법

(단, 마지막+1 번째와의 차를 구할 수 없으니, 이 값은 구하지 않는다)

# 다른 언어에서는 len과 index를 이용하여 각 원소에 접근
# 파이썬에서는 zip을 이용하면 index를 사용하지 않고 원소에 접근할 수 있음 

# number1에는 mylist[0]부터, number2에는 mylist[1]부터 iterate 한다
# 또한, zip 함수에 서로 길이가 다른 리스트가 인자로 들어오는 경우 길이가 짧은 쪽까지만 이터레이션이 이루어 진다.
# 때문에 mylist[1:] 이 mylist 보다 길이가 짧아서 n 길이의 리스트에서 n,n+1 번째 차이를 구하지 않게 된다.

def solution(mylist):
    answer = []
    for number1, number2 in zip(mylist, mylist[1:]):
        answer.append(abs(number1 - number2))
    return answer

if __name__ == '__main__':
    mylist = [83, 48, 13, 4, 71, 11]    
    print(solution(mylist))

 

8. 모든 멤버의 type 변환하기 : map

# 파이썬
# map(변환할 type, 변환할 대상)
list1 = ['1','100','33']
list2 = list(map(int, list1))

 

9. Sequence 멤버를 하나로 이어 붙이기 : join

* Ex) ['1','100','33'] 를 이어 붙여 문자열 '110033' 만들기

# 다른 언어에서는 for문으로 원소를 하나씩 이어 붙임
# 파이썬

mylist = ['1','100','33']
answer = ''.join(mylist)

 

10. 곱집합(Cartesain product) 구하기 

* Ex) 'ABC', 'xy' 의 곱집합은 Ax Ay Bx By Cx Cy 

# 다른 언어에서는 for문을 이용하여 두 iterable의 원소를 하나씩 곱함
# 파이썬에서는 itertools.product 를 이용하여 for문 없이 곱집합을 구할 수 있음

import itertools

iter1 = 'ABCD'
iter2 = 'xy'
iter3 = '1234'

print(list(itertools.product(iter1,iter2,iter3)))
# itertools.product 사용하면 튜플로 저장됨

 

11. 2차원 리스트를 1차원 리스트로 만들기

# 파이썬

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

# 방법1 : sum 함수
answer = sum(mylist,[])

# 방법2 : list comprehension
answer = [element for array in mylist for element in array]

 

12. 순열과 조합 : itertools.permutations() 사용

#파이썬 

import itertools

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

 

13. for문과 if문을 한번에 : List comprehension 의 if문

# 파이썬

# 주어진 list에서 짝수인 수만 제곱하여 리스트에 추가
mylist = [3,2,6,7]
answer = [number**2 for number in mylist if number%2==0]

 

14. for-else문 

* Ex) 자연수 5개가 주어질 때, 숫자를 차례로 곱해 나온 수가 제곱수가 되면 'found'를 출력하고, 모든 수를 곱해도 제곱수가 나오지 않으면 'not found' 를 출력하는 코드 작성 시

# 파이썬
# 숫자 x 가 제곱수 인지 확인하려면 sqrt(x) == int(sqrt(x)) 인지 확인 
# for-else 문은 for문이 중간에 break 되지 않으면 실행
# 때문에 found 하면 break 되어 else문은 실행하지 않고 found 하지 못하면 not found 실행

import math

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

 

15. 두 변수의 값 바꾸기

a = 3
b = 'abc'

a,b = b,a

 

16. 이진 탐색 : bisect.bisect() 사용

# 파이썬

import bisect

mylist = [1,2,3,7,9,11,33]
print(bisect.bisect(mylist,3))        # 해당 값의 뒷 인덱스 반환
print(bisect.bisect_left(mylist,3))   # 해당 값의 인덱스 반환
print(bisect.bisect_right(mylist,3))  # 해당 값의 뒷 인덱스 반환