파이썬에서 여러 영단어로 구성된 리스트를 만들어야 할 필요가 있습니다.
저는 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()을 쓰는게 더 비효율적일 수 있지만 칼럼이 많은 경우라면 아주 유용하답니다.
위의 코드를 귀차니즘을 최고 미덕으로 삼는 진정한 프로그래머들에게 바칩니다~
'Programming' 카테고리의 다른 글
{Python} 엑셀 DDE 실시간 데이터를 ML/DL 엔진과 연결하는 방법 (2) | 2019.12.30 |
---|