itertools 모듈

asda주asda·2022년 1월 30일
0

Python

목록 보기
17/31

chain()

여러 개의 리스트를 연결하는데 있어 +연산자도 있지만, itertools.chain()를 사용할 수도 있다.

from itertools import *
test = [1,2,3]
test1 = [['사람','동물','식물'],['김춘자','옥춘복','임복순']]
test2 = [('hello', 'man'),'안녕']

result =list(chain(test, test1, test2))
print(result)

출력 결과:
[1, 2, 3, ['사람', '동물', '식물'], ['김춘자', '옥춘복', '임복순'], ('hello', 'man'), '안녕']

위와 같이 리스트들의 요소들을 연결하여 반환해준다.

repeat()

itertools.repeat(list, n)의 형태의 repeat 메서드는 리스트를 n번 만큼 반복 하여 주소를 반환해준다.

from itertools import *
test = [1,2,3]
test1 = [['사람','동물','식물'],['김춘자','옥춘복','임복순']]
test2 = [('hello', 'man'),'안녕']

result = repeat(test,3)
print(list(result))

출력 결과:
[[1, 2, 3], [1, 2, 3], [1, 2, 3]]

0개의 댓글

관련 채용 정보