CSV File Row extraction using Python

Question

My aim is to have a user, enter a value into a tkinter GUI. This value is then used to extract a row that contains the users input value. Example of test.CSV file;

The user input is 1000, thus
[(field1 cat),(field2 fred),(field3 apple),(field4 1000),(field5 car)] will be extracted

I have achieved the CSV lookup by using the following code

import csv

with open('test.csv') as f:
    reader =csv.DictReader(f)
    rows = [row for row in reader if row['field4'] =='1000']
for row in rows:
    print (row)

The next step is to print only the columns that i require i.e. not all the columns are required :

I only want [(field1 cat), (field3 apple)], can this be achieved using the above code, or is there a better way of a Vlookup() in Python ?

use pandas, df = pd.read_csv("ur_file"). df = df[df.field4 == 1000] — Khalil Al Hooti
– Khalil Al Hooti, Commented Aug 22, 2018 at 15:11

Hans Musgrave · Accepted Answer · 2018-08-23 04:39:11Z

1

The standard tool in Python for working with tables is pandas, and its core tool is an object called a DataFrame.

import pandas as pd

df = pd.read_csv("test.csv")
print(df[["field1", "field2"]][df["field4"]==1000])

Update: As requested in a comment, you may or may not want the headers printed. By explicitly calling the .to_string() method for the DataFrame you can avoid that pretty easily.

print(df[["field1", "field2"]][df["field4"]==1000].to_string(header=False))

edited Aug 23, 2018 at 4:39

answered Aug 22, 2018 at 15:10

Hans Musgrave

7,1912 gold badges20 silver badges40 bronze badges

Sign up to request clarification or add additional context in comments.

3 Comments

John Over a year ago

Is there a way to print without displaying the column headers ?

John Over a year ago

I also found another solution .tolist() - thanks for all your help !!

Hans Musgrave Over a year ago

Glad that works. Be a little careful if the dataframe is too large since that will instantiate another object (your list) that's essentially a copy of the dataframe before creating the string it prints.

Collectives™ on Stack Overflow

CSV File Row extraction using Python

1 Answer 1

3 Comments

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

3 Comments

Related