본문 바로가기

Programming

{Python} 단어로 구성된 긴 list를 손쉽게 만드는 방법

파이썬에서 여러 영단어로 구성된 리스트를 만들어야 할 필요가 있습니다.

 

저는 pandas DataFrame의 columns를 지정할 때 이런 경우가 많습니다.

 

아래처럼 헤더가 없는 csv파일을 읽어온 후 제가 원하는 대로 칼럼명을 지정해야 하는 경우죠. 

 

import pandas as pd

df = pd.read_csv('data.csv')

df.columns=['date', 'item', 'value1', 'value2', 'value3']  # 에고 귀찮다..

 

전 따옴표 열고 닫고 콤마 타이핑 하는 반복적인 작업을 하기 싫어서 아래 처럼 코딩 한답니다.

 

import pandas as pd

df = pd.read_csv('data.csv')

df.columns = 'date item value1 value2 value3'.split()  # 문자열을 리스트로!

 

딱 보시면 아시겠지만 문자열을 공백 기준으로 잘라낸후 리스트로 반환하는 split 함수를 이용하였습니다.

 

위의 예시에는 칼럼이 5개라 split()을 쓰는게 더 비효율적일 수 있지만 칼럼이 많은 경우라면 아주 유용하답니다.

 

위의 코드를 귀차니즘을 최고 미덕으로 삼는 진정한 프로그래머들에게 바칩니다~