How to take inputs from a CSV file and write a specific output with Python?

Question

I have a CSV file with data like the following:

a,b,c,d,e,f
0,0,AER,0,DME,0
0,0,ASF,0,LED,0

How do I take inputs from columns C and E, and output them into something like:

I like [column C] and [column E]
I like [column C] and [column E]

example:

I like AER and DME
I like ASF and LED

Right now I have the following code:

import csv

header1 =['c']
header2 =['e']

with open('routes2.csv', 'rb') as csvfilein, open('out.csv', 'wb') as csvfileout:
    reader = csv.DictReader(csvfilein)
    writer1 = csv.DictWriter(csvfileout, header1, extrasaction='ignore')
    writer2 = csv.DictWriter(csvfileout, header2, extrasaction='ignore')
    for line in reader:
        writer1.writerow(line), writer2.writerow(line)

I'm stuck at trying to figure out how to append the text to data from columns C and E. How would I do this?

The output you desire isn't CSV - so I've answered based on just creating a line that can be used — Jon Clements
– Jon Clements, Commented Oct 14, 2013 at 15:00

Jon Clements · Accepted Answer · 2013-10-14 14:57:02Z

4

You can use string formatting and provide the row object returned by csv.DictReader, eg:

with open('routes2.csv', 'rb') as csvfilein:
    reader = csv.DictReader(csvfilein)
    for row in reader:
        print 'I love {c} and {e}'.format(**row)

answered Oct 14, 2013 at 14:57

Jon Clements

143k34 gold badges254 silver badges288 bronze badges

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

1 Comment

Infection Over a year ago

Sorry, I should've been clearer in my question! I was looking to output I love {c} and {e} into a separate file. Thanks Jon!

cdonts · Accepted Answer · 2013-10-14 15:05:41Z

1

Like this?

with open('routes2.csv', 'rb') as csvfilein:
reader = csv.DictReader(csvfilein)
for line in reader:
    print "I like %s and %s" % (line["c"], line["e"])

Output:

I like AER and DME
I like ASF and LED

answered Oct 14, 2013 at 15:05

cdonts

9,7116 gold badges53 silver badges78 bronze badges

Comments

martineau · Accepted Answer · 2013-10-14 16:15:39Z

1

Your output file consists of just plain text, is therefore not a .csv file, so there's no need to use a csv.DictWriterto create it. It's also easy to redirect aprintstatement's output to a file as illustrated.

import csv

header1 = ['c']
header2 = ['e']
format_specifier = 'I like %({0[0]})s and %({1[0]})s'.format(header1, header2)

with open('routes2.csv', 'rb') as csvfilein, open('out.txt', 'w') as fileout:
    for row in csv.DictReader(csvfilein):
        print >> fileout, format_specifier % row

answered Oct 14, 2013 at 16:15

martineau

124k29 gold badges181 silver badges319 bronze badges

Comments

Vivek · Accepted Answer · 2013-10-14 15:07:00Z

0

Try:

import csv
    header1 =['c']
    header2 =['e']

    with open(r'<input_file_path>', 'rb') as csvfilein, open(r'<output_file_path>', 'wb') as csvfileout:
        reader = csv.DictReader(csvfilein)
        for line in reader:
            csvfileout.write("I like "+line.get(header1[0])+" and "+line.get(header2[0])+"\n")

OUTPUT:

I like AER and DME

I like ASF and LED

answered Oct 14, 2013 at 15:07

Vivek

9102 gold badges10 silver badges27 bronze badges

1 Comment

Infection Over a year ago

Thanks Vivek, that works! Now I understand how this works better. :)

Collectives™ on Stack Overflow

How to take inputs from a CSV file and write a specific output with Python?

4 Answers 4

1 Comment

Comments

Comments

1 Comment

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

4 Answers 4

1 Comment

Comments

Comments

1 Comment

Your Answer

Sign up or log in

Post as a guest

Related