기존의 list의 값을 이용해서 새로운 list를 만드는 방법 list comprehension

아래 방법을 알기 전에 내가 기존에 사용하던 방식

배열을 선언해준 후 각 값에 2를 곱해서 배열에 담기 때문에 코드가 많이 지저분하다.

my_list = [1, 3, 5, 7, 9]
squares = []

for x in my_list:
    squares.append(x * 2)
    
print(squares)

 

위의 과정을 일부 생략하고 python문법으로 작성한 코드

my_list = [1, 3, 5, 7, 9]
squares = [x * 2 for x in my_list]

print(squares)

 

 

 

위의 방법을 selenium에서 다음과 같이 활용할 수 있다.

# 기존의 코드
number_list = []
for number_link in number_link_list:
    number = number_link.get_attribute("textContent")
    number_list.append(claim_number)
    
    
# 수정한 코드
number_list = [
    number_link.get_attribute("textContent") for number_link in number_link_list
]

기능은 동일함