Python:从csv中提取特定行作为列表

时间:2018-06-27 10:22:17

标签: python csv parsing

可能是转发,但我找不到适合我的情况的解决方案。

我有一个.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”,因此该解决方案似乎不起作用

1 个答案:

答案 0 :(得分:2)

使用熊猫将csv文件读入数据框

import pandas as pd
sourceinput = pd.read_csv('myfile.csv')

outputlist = sourceinput['id'].loc[sourceinput['id'] == <value_you_need>].tolist()