Python读取csv文件中的某一列

比如这样的一个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

猜你喜欢

转载自blog.csdn.net/qq_43523725/article/details/104279120