在python中将数据写入CSV文件

Question

我下面的代码使用正则表达式检查邮政编码的有效性。 邮政编码以数组形式给出。 for循环使用给定的正则表达式检查每个邮政编码的有效性。 现在，我想将所有有效的邮政编码写入csv文件。 下面是我的代码：

import csv
import re

regex = r"(GIR\s0AA)|((([A-PR-UWYZ][0-9][0-9]?)|(([A-PR-UWYZ][A-HK-Y][0-9]((BR|FY|HA|HD|HG|HR|HS|HX|JE|LD|SM|SR|WC|WN|ZE)[0-9])[0-9])|([A-PR-UWYZ][A-HK-Y](AB|LL|SO)[0-9])|(WC[0-9][A-Z])|(([A-PR-UWYZ][0-9][A-HJKPSTUW])|([A-PR-UWYZ][A-HK-Y][0-9][ABEHMNPRVWXY]))))\s[0-9][ABD-HJLNP-UW-Z]{2})"

postcodes = ['$%±()()','XX XXX','A1 9A','LS44PL','Q1A 9AA','V1A 9AA','X1A 9BB','LI10 3QP','LJ10 3QP','LZ10 3QP','A9Q 9AA','AA9C 9AA','FY10 4PL','SO1 4QQ','EC1A 1BB','W1A 0AX','M1 1AE','B33 8TH','CR2 6XH','DN55 1PT','GIR 0AA','SO10 9AA','FY9 9AA','WC1A 9AA']


for x in postcodes:
    if(re.findall(regex,x)):
        with open('test2.csv','w',newline='') as fp:
            a = csv.writer(fp)
            a.writerows(x)

该代码的问题是它没有将所有有效的邮政编码写入csv文件，而是仅以以下格式写入了最后一个邮政编码（WC1A 9AA）：

W
C
1
A 

9
A
A

我不知道我在哪里犯错。 请帮忙。

Answer 1

那里有几个问题，但最大的问题是'w' -每次写入文件时都会将其清除！ :)将其更改为'a'以进行追加。

其次，我不确定您要做什么，但是如果您想将它们全部写在单独的行上

codes = []
for x in postcodes:
    if(re.findall(regex,x)):
       codes.append([x])

with open('test2.csv','w',newline='') as fp:
    a = csv.writer(fp)
    a.writerows(codes)

Answer 2

通过在循环中设置带有标志“ w”的打开文件命令，它将在每次迭代中删除并创建一个新文件，因此，您只会得到最后的邮政编码。 这应该为您提供正确的结果：

with open('test2.csv','w',newline='') as fp:
    for x in postcodes:
        if(re.findall(regex,x)):
            a = csv.writer(fp)
            a.writerows(x)

是的，我忘了。 “ fp.close（）”与“ with open”语句无关。

Answer 3

如@jcfollower所建议的那样，在open函数中将'w'更改为'a'模式以附加而不是覆盖每个循环中的文件，或仅将with上下文管理器中的for循环移动到with上下文管理器中。