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

旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。

更多推荐