如何仅将包含特定字符串的行（在CSV文件中）另存为单独的CSV文件？

Question

I want to save only lines containing a particular String (in a CSV file) as a separate CSV file?

Consider the following file:

foo,bar,zig,zag
zog,bar,zag,zag
qux,zig,zag,zag
quuux,bar,zag,zag

When filtered for bar, the result should be a separate CSV file:

foo,bar,zig,zag
zog,bar,zag,zag
quuux,bar,zag,zag

What's the most efficient way to achieve this? Is there code to do this, as I need to do it for about 200 such names as "bar".

Answer 1

You could just filter each row by a certain column (assuming it's always the same column):

import csv

names = ['bar', 'foo', 'foobar']
data = []

# read data
with open('all.csv', 'r') as csvfile:
    spamreader = csv.reader(csvfile, delimiter=',')
    for row in spamreader:
        if row[1] in names:
            data.append(row)

# maybe some more processing
# ...

# write back data that meets criteria
with open('bar.csv', 'wb') as csvfile:
    spamwriter = csv.writer(csvfile, delimiter=',')
    spamwriter.writerows(data)