짝꿍 되는 경우의 수(combinations), 사다리 타기 파이썬 코드

2024. 2. 18. 23:56전처리/PYTHON

 

01 짝꿍이 되는 경우의 수 나열 (중복 제거)


import itertools # 경우의수

data = ['홍진경', '조세호', '김숙', '장우영']
num = 0
print("Q. 홍김동전 파트너가 될 경우의 수를 구한다면?", end = "\n\n")

for a, b in itertools.combinations(data, 2):
    num += 1
    print("{}. {}과 {} 둘은 짝꿍이 될 수 있다.".format(num, a, b))

 

02 사다리 타는 기능 코드 (역할 분배)


import random, itertools
data = ['홍진경' , '김숙', '장우영', '조세호', '주우재']
role = ['발표', '피피티', '자료조사']
new_data = random.sample(data, len(data)) # 어차피 리스트+리스트니까 뽑지말고 리스트만 섞음 편하네
reslut = itertools.zip_longest(new_data, role, fillvalue = '데이터분석') # result라는 객체는 itertools 클래스의 인스턴스임.
list(reslut)