파이썬 pandas csv 파일 읽어서 특정 열 값 가져오기

파이썬 pandas를 이용해 csv파일을 읽고 해당 dataframe을 이용해 특정 열의 값을 가져올 경우 사용한다.

 

설치 :

 

-m pip install pandas

 

import pandas as pd

tickerList = pd.read_csv("tickerList.csv", names=['name','ticker'], encoding='utf-8')
ticker = tickerList.loc[(tickerList['name'] == '삼성전자'), 'ticker'].values
print(ticker)

read_csv는 pandas를 이용해 csv파일을 읽을 수 있는 함수다. names는 읽어들인 데이터의 열 이름을 첫 열부터 차례로 지정한다.

 

tickerList의 loc를 이용해 원하는 열과 행의 데이터를 찾을 수 있다. 대괄호의 첫번째 데이터는 행에 대한 정보를 두번째 데이터인 'ticker'는 열에대한 데이터를 지정한다. 행과 열을 만족하는 데이터를 찾고 'values'를 이용해 해당 테이블의 변수를 반환한다.

 

결과:

['5930']