python csv find text in file and then find a specific column

Question

i have a database of csv something like

apple,10,red,date
orange,12,yellow,date
pear,13,green,date
pineapple,14,brown,date

i want to first search for eg: pineapple and then get third column" from it, i need some ideas, i am using the python csv module

i am able to get

#!/usr/bin/env python


import csv
import sys

file = open("/tmp/database.txt", 'rt')
reader = csv.reader(file)
for i in reader:
   print i

output:

['apple', '10', 'red', 'date']
['orange', '12', 'yellow', 'date']
['pear', '13', 'green', 'date']
['pineapple', '14', 'brown', 'date']

phihag · Accepted Answer · 2011-12-15 12:47:15Z

6

You can just check whether the first element of the line is pineapple. A neat way to get all the third values is to use a list comprehension:

import csv
with open("/tmp/database.txt", 'r') as file:
  reader = csv.reader(file)
  third_where_pineapple = [line[2] for line in reader if line[0] == 'pineapple']
  print (third_where_pineapple)

answered Dec 15, 2011 at 12:47

phihag

289k75 gold badges475 silver badges489 bronze badges

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

Collectives™ on Stack Overflow

python csv find text in file and then find a specific column

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