[Python / 기본문법] 자주 쓰는 파이썬 라이브러리

순열과 조합

#순열과 조합
from itertools import permutations

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

result = list(permutations(data, 3))  #모든 순열 구하기
print(result)
#[('A', 'B', 'C'), ('A', 'C', 'B'), ('B', 'A', 'C'), ('B', 'C', 'A'), ('C', 'A', 'B'), ('C', 'B', 'A')]

from itertools import combinations

result = list(combinations(data, 2))  #2개를 뽑는 모든 조합 구하기
print(result)
#[('A', 'B'), ('A', 'C'), ('B', 'C')]

 

Couter

#Counter
from collections import Counter

counter = Counter(['red', 'blue', 'red', 'green', 'blue', 'blue'])  #객체 선언

print(counter['blue']) #3
print(counter['red']) #2
#사전 자료형으로 반환
print(dict(counter)) #{'red': 2, 'blue': 3, 'green': 1}

 

최대공약수와 최소공배수

#최대공약수와 최소공배수
import math


def lcm(a, b):
    return a * b // math.gcd(a, b)

a = 21
b = 14

#죄대공약수
print(math.gcd(21, 14)) #7
#최소공배수
print(lcm(a, b)) #42