[Python] 리스트 내포(Comprehension)

For과 If문을 비교적 쉽고 간단하게 사용할 수 있는 방법

 

For문

nums = [i for i in range(5)]
print(nums)
#[0,1,2,3,4]

nums = [100,200,300]
double_nums = [i*2 for i in nums]
print(double_nums) #[200,400,600]

If문

nums = [i for i in range(10) if i % 2 == 0]
print(nums) # [0,2,4,6,8]

nums = [100,200,300,400,500]
double_nums = [i * 2 in range nums if i >= 300]
print(double_nums)  #[600, 800, 1000]

 

# 리스트 내포
# for 사용
nums = [i for i in range(5)]
print(nums) #[0, 1, 2, 3, 4]

nums2 = [100,200,300]
double_nums2 = [i * 2 for i in nums2]
print(double_nums2) # [200, 400, 600]


# if 사용
nums3 = [i for i in range(10) if i % 2 == 0]
print(nums3) # [0, 2, 4, 6, 8]

nums4 = [100,200,300,400,500]
double_nums4 = [i * 2 for i in nums4 if i >= 300]
print(double_nums4) # [600, 800, 1000]

 

실습

word_list = ['apple','watch','apolo','star','abocado']

#리스트 내포 사용
words = [i for i in word_list if i[0] =='a']
print(words)

#리스트 내포 사용 안함
for word in word_list:
    if word[0] != 'a':
        word_list.remove(word)



print(word_list)


words_list = ['오메가3',None,'비타민C500',None,"홍상절편"]


# 리스트 내포 미사용
result = []
for word in words_list:
    if word == None:
       result.append("")
    else:
        result.append(word)

print(result)


# 리스트 내포 사용 시
# words = ["" if i == None else i for i in words_list]
# print(words)