python - csv写入错误

Question

我正在从单个 csv 文件(data_labelled.csv)写入不同的 csv 文件基于第 32 列(它包含不同的节点名称)，我想要每个节点都有一个单独的文件。

csv 文件:

2040696 58.475039683    4.6011785714    4.9225238095    1.3880952381    550.2218254 968424.34127    1046524 job_1456818565071_0001  task_1456818565071_0001_r_000000    REDUCE  aditi   0
2040692 56.9    5.89    6.12    2   543.66666667    1045940 1046524 job_1456990723980_0001  task_1456990723980_0001_r_000000    REDUCE  kalpesh 0
2040692 56.9    4.5633333333    5.2066666667    3   544 1045940 1046524 job_1456990723980_0001  task_1456990723980_0001_r_000000    REDUCE  kalpesh 0
2040692 56.9    3.9 4.75    3   544 1045940 1046524 job_1456990723980_0001  task_1456990723980_0001_r_000000    REDUCE  kalpesh 0
2040696 57.1    8.63    10.356666667    19.666666667    573 1046524 1046524 job_1456990723980_0001  task_1456990723980_0001_m_000000    MAP pooja   0
2040696 57.1    11.82   14.55   29  626 1046524 1046524 job_1456990723980_0001  task_1456990723980_0001_m_000000    MAP pooja   0
2040696 57.1    11.82   14.55   29  626 1046524 1046524 job_1456990723980_0001  task_1456990723980_0001_m_000000    MAP pooja   0

根据代码，应该有三个不同的文件 pooja.csv 、 kalpesh.csv 、 aditi.csv 但只有 kalpesh.csv 正确完成，其他两个是空的。

代码:

file = csv.reader(open('data_labelled.csv','r'))
print node_list
for node in node_list:
    filename = '{}.csv'.format(node)
    file2_obj=open(filename,'w')    
    file2 = csv.writer(file2_obj)
    entries=list(file)
    for entry in entries:
        if(entry[32] == node):
            file2.writerow(entry)
    file2_obj.close()

node_list 的结果很好:

[u'kalpesh', u'aditi', u'pooja']

但是为什么另外两个文件是空的？出了什么问题？

Answer 1

将 entries=list(file) 放入循环之外，这就是问题所在。

您的代码应如下所示。

file = csv.reader(open('data_labelled.csv','r'))
entries=list(file)
print node_list
for node in node_list:
    filename = '{}.csv'.format(node)
    file2_obj=open(filename,'w')    
    file2 = csv.writer(file2_obj)
    for entry in entries:
        if(entry[32] == node):
            file2.writerow(entry)
    file2_obj.close()