How to easily filter from csv with python?

Question

Assuming I have the following CSV file:

User ID   Name        Application  

001       Ajohns      ABI
002       Fjerry      Central
900       Xknight     RFC
300       JollK       QDI
078       Demik       Central

Is there some easy way to (import this into some data structure)? and/or be able to easily perform the following operations in python:

1) Get all user IDs with Application=Central
2) Get the row where name="FJerry" then extract say the "userid" value from
3) Give me the filtered rows for those with "Application=Central" so I can write out to CSV
4) Give me all the rows where Name="Ajohn", and Application="ABI" such that I could easy do a len() to count them up?

Is there some python library or what is the easiest way to accomplish the above?

Have you looked at the csv module?

Fred Larson
– Fred Larson

2012-10-30 04:47:22 +00:00
Commented Oct 30, 2012 at 4:47 — Fred Larson
– Fred Larson, Commented Oct 30, 2012 at 4:47

Burhan Khalid · Accepted Answer · 2012-10-30 05:12:17Z

1

Trivial using DictReader. You need to pass excel-tab as the dialect since your fields are tab delimited.

rows is a list of dictionaries.

>>> with open(r'd:\file.csv','rb') as f:
...     reader = csv.DictReader(f,dialect='excel-tab')
...     rows = list(reader)
...
>>> [x['User ID'] for x in rows if x['Application'] == 'Central']
['002', '078']

answered Oct 30, 2012 at 5:12

Burhan Khalid

175k20 gold badges254 silver badges291 bronze badges

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

Collectives™ on Stack Overflow

How to easily filter from csv with python?

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related