column drop ( 사용하는것만 재지정 )

potato·2021년 8월 12일
0

파이썬

목록 보기
13/14
import pandas as pd
import os

fname = os.getenv('HOME') + '/xxx.tsv'

col_names = ['user_id', 'artist_MBID', 'artist', 'play']   # 전체 컬럼명
data = pd.read_csv(fname, sep='\t', names= col_names)
data.head(10)

# 사용하는 컬럼만 남겨줌
using_cols = ['user_id', 'artist', 'play']
data = data[using_cols]
data.head()
profile
안녕하세요~

0개의 댓글