[python] 자주 사용하는 주요 라이브러리

김우경·2020년 11월 8일
0

내장함수

#리스트의 원소 합
result = sum([1,2,3,4,5])

#리스트의 최소값
result = min([1,2,3,4,5])

#리스트의 최대값
result = max([1,2,3,4,5])

#문자열 수식 계산
result = eval("(3+5)*7")

itertools

  • permutations

    : r개의 데이터를 뽑아서 일렬로 나열하는 모든 경우

  • combinations

    : r개의 데이터를 뽑아서 순서 상관없이 나열하는 모든 경우

  • product

    : r개의 데이터를 뽑아서 일렬로 나열하는데 중복 허용

  • combinations_with_replacement

    : r개의 데이터를 뽑아서 순서 상관없이 나열하는데 중복 허용

from itertools import permutations, combinations, product, combinations_with_replacement

data = ['A', 'B', 'C']

perm = list(permutations(data, 3))
[('A', 'B', 'C'), ('A', 'C', 'B'), ('B', 'A', 'C'), ('B', 'C', 'A'), ('C', 'A', 'B'), ('C', 'B', 'A')]

comb = list(combinations(data, 3))
[('A', 'B', 'C')]

prod = list(product(data, repeat = 2))
[('A', 'A'), ('A', 'B'), ('A', 'C'), ('B', 'A'), ('B', 'B'), ('B', 'C'), ('C', 'A'), ('C', 'B'), ('C', 'C')]

comb_w = list(combinations_with_replacement(data, 2))
[('A', 'A'), ('A', 'B'), ('A', 'C'), ('B', 'B'), ('B', 'C'), ('C', 'C')]

bisect

  • 정렬된 배열에서 특정 원소 찾을때

    → O(logN)에 동작

  • bisect_left(a, x)

    : 정렬된 순서를 유지하면서 리스트 a에 데이터 x 삽입할 가장 왼쪽 인덱스 찾기

  • bisect_right(a, x)

    : 정렬된 순서를 유지하면서 리스트 a에 데이터 x 삽입할 가장 오른쪽 인덱스 찾기

    → 값이 특정 범위에 속하는 원소의 개수

from bisect import bisect_right, bisect_left

a = [1,2,3,4,4,8]
x = 4

print(bisect_left(a,x))
print(bisect_right(a,x))
  • count_by_range(a, left_value, right_value)

    : 정렬된 리스트 a 에서 left_value≤x≤ right_value인 x개수를 O(logN)으로

    from bisect import bisect_right, bisect_left
    
    def count_by_range(a, left_value, right_value):
        rindex = bisect_right(a, right_value)
        lindex = bisect_left(a, left_value)
        return rindex-lindex
    
    a = [1,2,3,4,4,8]
    x = 4
    
    print(count_by_range(a,4,4))

collections

  • deque

    : 연속적으로 나열된 데이터의 시작/끝에 데이터 삽입/삭제시 O(1)

    ↔ list는 삽입, 삭제, 인덱싱, 슬라이싱 등 가능하지만 삭제시 O(N)

    • 삽입 : append(), appendleft()

    • 삭제 : pop(), popleft()

      → deque를 queue로 사용 ?

      : 삽입은 append(), 삭제는 popleft()

  • Counter

    : 등장 횟수를 세는 기능 → iterable 객체 내에 원소가 몇번 등장했는지

    from collections import Counter
    
    counter = Counter(['red', 'blue', 'red', 'green', 'blue'])
    
    print(counter['red'])
    print(counter['blue'])
    print(counter['green'])
    
    print(dict(counter))
    #{'red': 2, 'blue': 2, 'green': 1}

math

  • factorial(x)
  • sqrt(x)
  • gcd(a, b)
profile
Hongik CE

0개의 댓글