Python: извлечение определенных строк из csv в виде списка - PullRequest
0 голосов
/ 27 июня 2018

Возможно, репост, но я не могу найти решение, которое работает в моем случае.

У меня есть файл .csv с идентификационным номером, связанным с такой строкой:

0   |   ABC
1   |   DEF
   ...
100 |   XYZ

Я хотел бы извлечь строку с идентификационным номером 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 голосов
/ 27 июня 2018

используйте pandas для чтения csv-файла в фрейм данных

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

outputlist = sourceinput['id'].loc[sourceinput['id'] == <value_you_need>].tolist()
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...