컴퓨터이야기

파이썬 for문 한줄 쓰기

서연유진맘 2023. 7. 6. 11:53

자주 생각이 안나서 적어놓는다.

list_a = [ j for i in range(4) for j in range(2)]

print(list_a)   /    [0, 1, 0, 1, 0, 1, 0, 1]

list_a = [ j for i in range(3) for j in range(6)]

print(list_a) / [0, 1, 2, 3, 4, 5, 0, 1, 2, 3, 4, 5, 0, 1, 2, 3, 4, 5]

list_b = [i for i in list_a if i > 3]

print(list_b) / [4, 5, 4, 5, 4, 5]