在python中的csv中将数据从一个工作表切割到另一个工作表

Question

我正在开发一个简单的程序来打开文件并读取某些行，然后将它们打印在另一个新文件中，但我想剪切它们并从早期的csv中删除它们。我怎么做？。这就是我的尝试。

import csv

f = open('1.csv')
csv_f = csv.reader(f)

content_value = []

for row in csv_f:
    if 'yepme' in row[2]:
        content_value.append(row)


g = open('output.csv', 'wb')

wr = csv.writer(g, dialect='excel')
wr.writerows(content_value)

我正在编辑并找到答案：

import csv

f = open('1.csv')
csv_f = csv.reader(f)

content_value = []
old_value = []
for row in csv_f:
    if 'yepme' in row[2]:
        content_value.append(row)
    else:
        old_value.append(row)

g = open('output.csv', 'wb')

wr = csv.writer(g, dialect='excel')
wr.writerows(content_value)

h = open('2.csv','wb')

ws = csv.writer(h, dialect='excel')
ws.writerows(old_value)

Answer 1

this question中提到了类似的问题。

简短解决方案：写两个文件：一个用提取的行，一个用剩余的。

编码解决方案：

import csv

with open('1.csv', 'r') as f:
    csv_f = csv.reader(f)

new_content = []
old_content = []

for row in csv_f:
    if 'yepme' in row[2]:
        new_content.append(row)
    else:
        old_content.append(row)

with open('output.csv', 'wb') as f:
    wr = csv.writer(f, dialect='excel')
    wr.writerows(new_content)

with open('1.csv', 'wb') as f:
    wr = csv.writer(f, dialect='excel')
    f.writerows(old_content)

我从未使用csv，但您应该明白这一点。如果您的csv文件非常巨大，您应该逐行读取和写入以避免内存问题。