Python Tip - by KSW

Swift Tip : https://velog.io/@heyksw/Swift-%EB%A1%9C-%EC%95%8C%EA%B3%A0%EB%A6%AC%EC%A6%98-%ED%95%B4%EA%B2%B0-%ED%8C%81

Contents

1. 리스트 컴프리헨션
2. 리스트 복사
3. 슬라이싱 활용
4. 간편한 이진탐색 bisect
5. replace 함수를 활용한, 다조건 문자열 파싱
6. lambda + sorted
7. 리스트에서의 중복 제거
8. 문자열과 리스트
9. 문자열 숫자/알파벳 판별 - isdigit(), isalpha()
10. 파이썬 for문의 index
11. pass와 continue의 차이
12. 빈 리스트일때의 return ... or 사용법
13. upper와 lower
14. in 활용법
15. return True or False 간략화
16. 문자열 곱셈 활용
17. heapq.heapify(myList)
18. zip 활용법
19. enumerate
20. max + lambda 와 any
21. product
22. 2차원 배열로 좌표평면 나타낼 때 주의점
23. deque의 잡기술
24. 16진법 이상의 n진수 변환
25. title(), capitalize() 를 활용한 문자열 다루기
26. 파이썬은 객체지향 언어이다.
27. defaultdict()
28. sum을 이용해서 2차원 배열 -> 1차원 배열로 변환하기
29. 정규표현식 re
30. 파이썬의 mutable / immutable
31. Counter, most_common()
32. 파이썬 for 문에서 조심할 것
33. 두 구간이 겹치는지 확인

1. 리스트 컴프리헨션

# 10 이하의 짝수를 담는 리스트 컴프리헨션  

list = [x for x in range(1,10+1) if x % 2 == 0]
--> list = [2,4,6,7,10]
   
   
# 리스트 컴프리헨션 내에 if와 else를 둘다 사용하고 싶다면 왼쪽에 작성해야한다.
# 짝수일 경우 그대로, 홀수일 경우 음수로 담는 코드   
   
listB = [ x if x % 2 == 0 else x * (-1) for x in range(1,10+1)]
--> listB = [-1,2,-3,4,-5,6,-7,8,-9,10]

# 0~4 의 수에서 짝수면 숫자를, 홀수면 odd를 담는 방법   

list = [x if x % 2 == 0 else 'odd' for x in range(5)]
#--> list = [0,odd,2,odd,4]

2. 리스트 복사

파이썬 리스트에서는 deep copy와 shallow copy가 일어난다.

listA = [1,2,3]
listB = listA
listB[0] = 100

위와 같이 코드를 작성한 경우
listA = [100,2,3] 으로 함께 변경된다.

따라서 제대로 된 복사(deep copy)를 하고 싶은 경우
listB = listA[:] 와 같이 코드를 작성해야 한다.

3. 슬라이싱 활용

listA = [1,2,3,4,5]를 역순으로 사용하고 싶다면
listB = listA[::-1]와 같이 활용하면 된다.

1,3,5와 같이 2칸씩 띄고 싶다면
listB = listB[::2]와 같이 활용하면 된다.

4. 간편한 이진탐색 bisect

Python에서는 bisect를 지원한다.
이진탐색의 시간복잡도는 O(log n) time이므로 코딩테스트에서 활용하기 좋다.

from bisect import bisect_left, bisect_right
array = [10,10,10,20,30,30,40,40,40,40,40,60,70,80,80,90]

cutLine = 30
index1 = bisect_left(array,cutLine)
index2 = bisect_right(array,cutLine)
print(index1,index2)

-> 출력 : 4 6
추가) 리스트에 없는 원소를 bisect 탐색한다면 ?

import bisect 
list = [10,20,30,40,50]
print(bisect.bisect_left(list,30))
print(bisect.bisect_right(list,30))

print(bisect.bisect_left(list,35))
print(bisect.bisect_right(list,35))

결과 :
2
3
3
3
따라서 bisect_left와 bisect_right의 값이 같은 경우, 리스트에 없는 원소라고 판단 할 수 있게 된다.

5. replace 함수를 활용한, 다조건 문자열 파싱

카카오 - 순위 검색 문제
'and'도 파싱해야 하고, ' '도 파싱해야 하는 상황
replace(' and ', ' ').split(' ')
-> ' and '를 ' '로 대체시킨 후, ' '를 기준으로 쪼갠다.

6. lambda + sorted

c = sorted(a, key = lambda x : x[0])

리스트 a에서 0번째 값을 key로 sorting 하는 방법.

7. 리스트에서의 중복제거

listA = [2,2,3,3,3,4,5,6,6,...]  # 정렬된 리스트
listA = list(set(listA)) # 중복제거
listA.sort()  # 재정렬

set을 이용해서 중복제거를 할 수 있지만, set을 사용하면 순서가 뒤섞이기 때문에 다시 sort()해줘야 한다.

8. 문자열과 리스트

문자열에서는 마치 리스트처럼 한 글자마다 번호를 새긴다.

x = 'banana'

--> x[0] = 'b'
--> x[2:4] = 'na'

하지만 문자열에 있는 글자를 리스트처럼 접근해서 변경할 수는 없다.

<pre>
<code>
x[0] = 'n' 
--> 불가능 !

꼭 바꾸고 싶다면 다음과 같이 할 수 있다.
방법 1)
x = 'n' + x[1:]
--> x = 'nanana'

방법 2)
x = list(x)
x[0] = n
x = "".join(x)

9. 문자열 숫자/알파벳 판별 - isdigit(), isalpha()

ex1 = '1'
ex2 = 'A'
print(ex1.isdigit(), ex2.isalpha())

결과 : True True

10. 파이썬 for문의 index

C++의 for문은
for(int i=0; i<10; i++)
처럼 쓰고 특정 조건일때 i의 값을 변경하면 조건문의 횟수를 조절할 수 있다.

하지만 Python에서는 for문 안에서 i 값을 변경하면 그 반복문 내에서는 반영되지만, 다음 반복시에는 원래의 i값(다음값)으로 돌아간다. 다음 예시는 1,2,3,4,5 를 가진 리스트에서, 3을 만날경우 건너뛰고 싶은 상황이다.

# 잘못된 예시
arr = [1,2,3,4,5]

for a in arr :
  if a == 3:
    a += 1
  print(a,end = ' ')
print()  # 결과 --> 1 2 4 4 5

# 해결법
a = 0
while a < len(arr):
  if arr[a] == 3:
    a += 1
  print(arr[a],end = ' ')
  a += 1
print()  # 결과 --> 1 2 4 5

11. pass와 continue의 차이

pass : 실행할 코드가 없는 것으로 다음 행동을 계속해서 진행한다.
continue : 바로 다음 순번의 loop를 수행한다.

12. 빈 리스트일때의 return ... or 사용법

answer = [...] 라는 리스트가 있을 때,

...
return answer or [-1]

다음과 같이 코드를 작성하면 answer가 비어있는 리스트일 경우 [-1]를 리턴한다.

13. upper와 lower

str = "Kim Sang Woo"
print(str.upper())
print(str.lower())

결과 : KIM SANG WOO, kim sang woo
참고로 isupper와 islower도 존재하고, 정렬하였을 때 대문자가 소문자보다 앞에 위치한다.

14. in 활용법

x 가 3,4,6,7 인 경우를 체크하고 싶을때, if와 elif를 반복할 필요없이

if x in [3,4,6,7]:

와 같이 코드를 작성하면 깔끔하다.

15. return True or False 간략화

예를들어, x가 짝수면 True, 홀수면 False를 리턴하는 함수를 생각해보자.

# 내가 짜던 코드
def func(x):
 return True if x % 2 == 0 else False

# 간략화
def func2(x):
 return x % 2 == 0

if문을 사용하지 않아도 True와 False를 판단하여 리턴한다.

16. 문자열 곱셈 활용

숫자가 나열된 문자열에서 마지막 4자리를 제외한 모든 숫자를 *로 변경하고 싶을 경우.

# 내가 전에 짰던 코드

def solution(phone_number):
    plist = list(phone_number)
    for i in range(len(plist)-4):
        plist[i] = "*"
    
    return "".join(plist)

# 문자열 곱셈을 이용하면 for문을 사용할 필요가 없어진다.

def solution(phone_number):
    
    return "*" * ( len(phone_number)-4 ) + phone_number[-4:]

17. heapq.heapify(myList)

정렬된 자료구조에서 min 값을 반복적으로 찾아야 하는 문제가 있다.
이러한 경우에는 heap 자료구조를 사용하는 것이 효율적이다.

myList = [....]
heapq.heapify(myList) # O(n)에 힙을 생성

# heapq.heappop() 을 하지않고 min 값만 구하는 방법
minValue = myList[0]

18. zip 활용법

sumList에 listA와 listB의 값들의 합을 pythonic 하게 담아보자.

# 전에 짜던 코드 방식
for i in range(len(listA)):
  sumList.append(listA[i]+listB[i])

# Pythonic
sumList = [x+y for x, y in zip(listA, listB)]

19. enumerate

19번과 20번 팁은 프로그래머스 Level2. 프린터 문제에서 공부한 내용이다.
enumerate는 “열거하다”라는 뜻이고, 리스트가 있는 경우 순서와 리스트의 값을 전달하는 기능을 가진다.

priorities = [2,1,3,2]
queue = [(i,p) for i, p in enumerate(priorities)]

queue = [(0,2), (1,1), (2,3), (3,2)]

20. max + lambda 와 any

max에서도 lambda 함수를 사용할수있다.
19번 팁에서 구한 queue에서, index 1을 기준으로 max값을 찾고싶다면, max(queue, key = lambda x : x[1])[1] 와 같이 코드를 작성하면된다.
max 바깥에 [1]이 또 붙음으로써, max에서 tuple의 1번째 index를 반환할 수 있다는 것이 핵심이다.

한편, queue에서 index 1기준으로 2보다 큰 tuple이 하나라도 존재하는지 판별하고 싶다.
이럴 땐 'any' 라는 키워드를 활용할 수 있다.

if any(2 < q[1] for q in queue) : return True

파이썬은 너무 아름답다..

21. product

데카르트 곱 이라고도 하는 cartesian product(곱집합)를 표현할 수 있다.

from itertools import product
myList = ["012","abc","!@#"]
print(list(product(*myList)))

--> [('0', 'a', '!'), ('0', 'a', '@'), ('0', 'a', '#'), ('0', 'b', '!'), ('0', 'b', '@'), ('0', 'b', '#'), ('0', 'c', '!'), ('0', 'c', '@'), ('0', 'c', '#'), ('1', 'a', '!'), ('1', 'a', '@'), ('1', 'a', '#'), ('1', 'b', '!'), ('1', 'b', '@'), ('1', 'b', '#'), ('1', 'c', '!'), ('1', 'c', '@'), ('1', 'c', '#'), ('2', 'a', '!'), ('2', 'a', '@'), ('2', 'a', '#'), ('2', 'b', '!'), ('2', 'b', '@'), ('2', 'b', '#'), ('2', 'c', '!'), ('2', 'c', '@'), ('2', 'c', '#')]

// product 로 모든 경우의 수 구하기   
cases = list(product([True, False], repeat=2))   
   
--> [(True, True), (True, False), (False, True), (False, Flase)]

22. 2차원 배열로 좌표평면 나타낼 때 주의점

먼저, 2차원 배열을 선언하기 위해서는 반드시 리스트 컴프리헨션을 사용해야한다.
가로의 길이가 8 (0<=x<8), 세로의 길이가 12 (0<=y<12)인 좌표평면을 표현하려면,

w = 8
h = 12
table = [[0] * h for _ in range(w)]
# h와 w의 순서를 잘 생각해볼 것

와 같이 선언해야, value = table[6][9] 와 같이 (6,9)에 접근할 수 있게된다.

23. deque의 잡기술

deque에는 maxlen, extend, extendleft, reverse, rotate 5가지의 유용한 잡기술이 존재한다.

from collections import deque

q = deque([1,2,3],maxlen = 4)

print(q)
q.extend([4,5,6]) # 오른쪽에 이어붙인다.
print('extend -',q)

q.extendleft([1,2,3]) # 1,2,3 을 순서대로 왼쪽에 이어붙인다.
print('extendleft -',q)

q.reverse() # 모든 원소 순서를 뒤집는다.
print('reverse -',q)

q.rotate(2) # 모든 원소를 2칸 오른쪽으로 회전한다.
print('rotate 2 -',q)

q.rotate(-1) # 모든 원소를 1칸 왼쪽으로 회전시킨다.
print('rotate -1 -',q)

출력 결과

deque([1, 2, 3], maxlen=4)
extend - deque([3, 4, 5, 6], maxlen=4)
extendleft - deque([3, 2, 1, 3], maxlen=4)
reverse - deque([3, 1, 2, 3], maxlen=4)
rotate 2 - deque([2, 3, 3, 1], maxlen=4)
rotate -1 - deque([3, 3, 1, 2], maxlen=4)

24. 16진법 이상의 n진수 변환

10 부터 'A'로 표기했을 때 n진수 변환 코드이다.

import string

tmp = string.digits+string.ascii_uppercase
def convert(num, base) :
    q, r = divmod(num, base)
    if q == 0 :
        return tmp[r] 
    else :
        return convert(q, base) + tmp[r]

25. title(), capitalize() 를 활용한 문자열 다루기

string = "HI my NAME is ksw"
print(string.title())
print(string.capitalize())

결과

Hi My Name Is Ksw
Hi my name is ksw

26. 파이썬은 객체지향 언어이다.

파이썬에서 class를 사용하여 객체지향 프로그래밍을 할 수 있다.
특이하게, self라는 키워드를 사용하는데, C++ 에서의 this와 비슷한 개념이라고 이해하면 된다.
클래스 안에서 메서드를 정의할 때는 첫 번째 인수를 항상 self로 지정해야 한다는 것을 외워야 한다.

27. defaultdict()

defaultdict()는 딕셔너리를 만드는 dict클래스의 서브클래스이다.
작동 방식은 dict()와 거의 동일한데, defaultdict()는 인자로 주어진 객체의 기본값을 딕셔너리 값의 초기값으로 지정할 수 있다.
숫자, 리스트, 셋 등으로 초기화 할 수 있기 때문에 여러 용도로 사용할 수 있다.

from collections import defaultdict
fruits = ['딸기','사과','참외','바나나','포도']
seasons = ['봄','여름','가을','겨울']
elems = ['봄','사과','참외','가을','겨울']

dic1 = dict()
dic2 = defaultdict(list)

for elem in elems:
  if elem in fruits:
    if not '과일' in dic1: dic1['과일'] = [elem]
    else: dic1['과일'].append(elem)
  elif elem in seasons:
    if not '계절' in dic1: dic1['계절'] = [elem]
    else: dic1['계절'].append(elem)

for elem in elems:
  if elem in fruits: dic2['과일'].append(elem)
  elif elem in seasons: dic2['계절'].append(elem)

print(dic1)
print(dic2)

{'계절': ['봄', '가을', '겨울'], '과일': ['사과', '참외']}   
defaultdict(, {'계절': ['봄', '가을', '겨울'], '과일': ['사과', '참외']})

28. sum을 이용해서 2차원 배열 -> 1차원 배열로 변환하기

arr = [[1],[2,3],[4,5,6]] 이라는 배열을 [1,2,3,4,5,6]으로 변경하고 싶을 때가 있다. sum() 함수를 Python 공식문서에서 깊게 파보면 sum(iterable, start=0)을 발견할 수 있다. start 시작 인자를 함께 더할 수 있는 것인데, 예를들어 a = [1,2,3]; sum(a,-1)을 실행하면 -1 + 1 + 2 + 3 = 5가 나오는 것이다. 이 사실을 이용해서 섹시하게 배열의 차원을 변경할 수 있다.

arr = [[1],[2,3],[4,5,6]]
result = sum(arr, [])

sum(arr,[])에서는 다음과 같은 식이 수행된다. [] + [1] + [2,3] + [4,5,6]
결과 : [1,2,3,4,5,6]

29. 정규표현식 re

파이썬은 re 라이브러리에서 정규표현식을 제공한다.
간혹 문자열 문제에서 엄청난 이득을 볼 수 있으니 알아두는 것이 좋다.

30. 파이썬의 mutable / immutable

Mutable : 가변적인 (int, string, tuple)
Immutable : 불가변적인 (list, dictionary, set)

# immutable
a = 1
b = "1"
c = (1,1)

# mutable
x = [1,1,1]
y = {'one': 1}
z = {1,2,3}

def change(a, b, c, x, y, z):
  a = 999
  b = "999"
  c = (999,999)
  x[0] = 999
  y['one'] = 999
  z.add(999)

change(a,b,c,x,y,z)
print(a,b,c,x,y,z)

31. Counter, most_common()

원소가 몇번 등장하는지 깔끔하게 정리할 수 있는 Counter 라는 클래스가 있다.
collections 모듈에 존재하며, dict 로 깔끔하게 형변환도 가능하다.

from collections import Counter
a = [10, 10, 10, 20, 40, 40]
counter = Counter(a)
print(counter)    # Counter({10: 3, 40: 2, 20: 1})
b = counter.most_common()
print(b)          # [(10, 3), (40, 2), (20, 1)]
c = counter.most_common(1)
print(c)          # [(10, 3)]

32. 파이썬 for문에서 조심할 것

파이썬 for 문은 간편한 기능도 제공하지만 주의할 점도 있다.

listA = [1,2,3,4,5]

for a in listA:
   a = 0

print(listA)   # -> [1,2,3,4,5] 바뀌지 않음

for i in range(len(listA)):
   listA[i] = 0
  
print(listA)   # -> [0,0,0,0,0] 바뀜

33. 두 구간이 겹치는지 확인

def is_overlap(a_start, a_end, b_start, b_end):
   if a_start <= b_end and b_start <= a_end:
      return True
   return False