[Python] Change element type in list

numpy나 torch 같은 경우 쉽게 type을 변경 할 수 있다.

하지만 python list에서 type을 변경해야 하는 상황이라면 어떻게 해야할까?

 

map(function, variable) 을 사용하자.

x = ['1','2','3','4','5']
print(list(map(float, x)))

print 결과

list()로 감싸지 않을 경우 map type의 객체가 출력되므로 자료구조형으로 감싸주어 출력하자.

 

다른 방법으론 list comprehension을 사용할 수 있다.

x = ['1','2','3','4','5']
x = [float(i) for i in x]

print(x)

print 결과

list comprehension은 loop뒤에 조건문을 추가하여 데이터 정제도 가능하다.

장점은 간편하게 표현할 수 있다는 점이지만 단점은 이중 loop와 같이 코드의 길이가 길어지는 경우 가독성이 떨어져 오히려 사용하지 않는 것이 나을 수도 있다. 각 상황에 맞게 적절하게 쓰는것이 중요하다.