可能是转发,但我找不到适合我的情况的解决方案。
我有一个.csv文件,其ID号与这样的字符串相关联:
0 | ABC
1 | DEF
...
100 | XYZ
我想提取一个ID号为x
的行并将其附加到列表中,因此理想情况是:
with open('myfile.csv', 'rb') as f:
reader = csv.reader(f)
results.append([row for idx, row in enumerate(reader) if idx[0] == x)])
尽管我虽然以字节模式打开它,但它却告诉我“ iterator should return strings, not bytes
”,因此该解决方案似乎不起作用
答案 0 :(得分:2)
使用熊猫将csv文件读入数据框
import pandas as pd
sourceinput = pd.read_csv('myfile.csv')
outputlist = sourceinput['id'].loc[sourceinput['id'] == <value_you_need>].tolist()