[파이썬] pandas, csv 특정 열 값을 리스트로 변경하기

아래와 같은 데이터가 있을 때 특정 열 값을 얻은 후 타입을 리스트로 변경하는 방법을 알아보자.

 

 

 

csv 파일 읽기

pandas 모듈을 사용할 것이기 때문에 먼저 pandas를 임포트해준 후 read_csv를 사용해 csv 파일을 읽어준다. read_csv를 사용해 읽는데 names=[]를 사용해 각 열의 이름을 지정해주고 한글 데이터의 경우 그냥 읽으면 깨지기 때문에 encoding='CP949'로 인코딩을 설정해준다.

 

import pandas as pd

csv = pd.read_csv('test.csv',\
		names=['gu','dong','name','category','microReview','commonAddr','addr'], \
		encoding='CP949')

 

 

특정 열 값을 리스트로 변경하기

그 다음, 속성 이름으로 특정 열을 선택해준다. 출력해보면, 객체 타입임을 알 수 있다.

 

cat = csv['category']

 

 

values를 사용해 선택해준 특정 열의 값만을 추출한다.

 

cat_val = cat.values

 

 

마지막으로, 추출한 값의 타입을 리스트로 변경하기 위해 tolist()를 수행해준다.

 

cat_list = cat_val.tolist()

 

 

당연히, values와 tolist()를 합쳐서 cat.values.tolist()를 해줘도 된다.

 

 

 

특정 열 값을 리스트로 변경하기, 전체 코드

전체 코드는 아래와 같다.

 

import pandas as pd

csv = pd.read_csv('test.csv',\
		names=['gu','dong','name','category','microReview','commonAddr','addr'], \
		encoding='CP949')

cat = csv['category']
print(cat.values.tolist())