Python读取csv文件中的某一列

Posted 2022-08-07 安之ccy

比如这样的一个csv文件：
想要读取imageID和label

注：imageID的第一个数据是：000858e87c40654a

import pandas as pd
path = 'yourfile.csv'
# 使用pandas读入
data = pd.read_csv(path) #读取文件中所有数据
# 按列分离数据
x = data[['ImageID', 'label']]#读取某两列
print(x)
y = data[['ImageID']]#读取某一列
print(y)

如果只读取一列数据，可以直接用y[i]访问

print(type(y)) #<class 'pandas.core.series.Series'>
print(len(y)) #1637
print(y[0])	#000858e87c40654a
print(type(y[0])) #str
print('000858e87c40654a' in y) #false