Iterieren durch bestimmte Zeilen in einer csvfile in Python

Question

Ich habe eine Programmieraufgabe, die csvfiles umfassen. Bis jetzt habe ich nur ein Problem damit, Werte nur von bestimmten Zeilen zu erhalten, welche die Zeilen sind, die der Benutzer nachschlagen möchte.

Als ich frustriert ich hängten nur jede Spalte in eine separate Liste, die sehr langsam ist (wenn die Liste für Test gedruckt wird), weil jede Spalte Hunderte von Werten hat.

Frage: die gewünschten Zeilen werden die Zeilen, deren Index [0] == USER_INPUT. Wie kann ich nur diese bestimmten Zeilen erhalten und die anderen ignorieren?

Answer 1

Sie Dies sollte eine Idee geben:

import csv 
with open('file.csv', 'rb') as f: 
    reader = csv.reader(f, delimiter=',') 
    user_rows = filter(lambda row: row[0] == user_input, reader) 

Answer 2

Python hat den Code, um das Modul csv

import csv 
rows=[] 
for row in csv.reader(open('a.csv','r'),delimiter=','): 
    if(row[0]==user_input): 
     rows.append(row) 

Answer 3

def filter_csv_by_prefix (csv_path, prefix): 
    with open (csv_path, 'r') as f: 
     return tuple (filter (lambda line : line.split(',')[0] == prefix, f.readlines())) 

for line in filter_csv_by_prefix ('your_csv_file', 'your_prefix'): 
    print (line)