search column, return row from excel using python

Question

I have a csv file with column A that has dates. I want to search the dates and return the corresponding row in array (VERY IMPT) format. How would I do that using python? Thank you.

excel file:

      A          B         C
1  12202014     403       302

2  12212014     312       674

3  12222014     193       310

input:

 Search csv file for 12212014

output:

[12212014,312,674]

attempt code:

date = '12212014'
with open('dates.csv', 'rt') as f:
     reader = csv.reader(f, delimiter=',')
     for row in reader:
          if date == row[0]: 
              print "found the date"

How do I return the row instead of just a print statement?

@PauloScardine Hello. I tried looking at that, but it does not show me how to search within a specific column and return the whole row in array format — user1681664
– user1681664, Commented Apr 1, 2014 at 2:25

alecxe · Accepted Answer · 2014-04-01 02:44:24Z

2

The basic idea is to create a dictionary/mapping date->line and get the value by the key:

import csv


data = {}
with open('test.csv', 'r') as f:
    reader = csv.reader(f)
    for line in reader:
        data[line[0]] = line

date_to_find = '12212014'
print data.get(date_to_find, 'Date not found')

prints:

['12212014', '312', '674']

This helps if you need the mapping afterwards to find the dates in it.

Another approach is to use generator expression to iterate over lines in a file until we find the date:

import csv


date_to_find = '12212014'
with open('test.csv', 'r') as f:
    print next((line for line in csv.reader(f) if line[0] == date_to_find), 
               'Date not found')

prints:

['12212014', '312', '674']

Hope that helps.

answered Apr 1, 2014 at 2:44

alecxe

476k127 gold badges1.1k silver badges1.2k bronze badges

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

Collectives™ on Stack Overflow

search column, return row from excel using python

1 Answer 1

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related