Python追加列标题&将列表中的列值追加到cs

Question

我正在尝试附加列标题（硬编码）并将列表中的列值附加到现有的csv。我没有得到想要的结果。在

方法1正在将结果附加到现有的csv文件上。方法2将现有csv的副本克隆到温度csv. 这两种方法都不能得到我想要的输出。在结果1中，它只在最后一行单元格之后追加。在结果2中，所有列表值都追加到每一行。预期的结果是我要找的。在

我在下面包含了我的代码。感谢任何意见或指导。在

现有CSV测试.csv

Type,Id,TypeId,CalcValues
B,111K,111Kequity(long) 111K,116.211768
C,111N,B(long) 111N,0.106559957
B,111J,c(long) 111J,20.061634

代码-方法1和方法2

^{pr2}$

方法1的结果

Type,Id,TypeId,CalcValues
B,111K,111Kequity(long) 111K,116.211768
C,111N,B(long) 111N,0.106559957
B,111J,c(long) 111J,20.0616340.1065599566767107
0.0038113334533441123
20.061623176440904

方法2的结果

Type,Id,TypeId,CalcValues,Results
B,111K,111Kequity(long) 111K,116.211768,0.1065599566767107,20.061623176440904,0.0038113334533441123
C,111N,B(long) 111N,0.106559957,0.1065599566767107,20.061623176440904,0.0038113334533441123
B,111J,c(long) 111J,20.061634,0.1065599566767107,20.061623176440904,0.0038113334533441123

预期结果

Type,Id,TypeId,CalcValues,ID
B,111K,111Kequity(long) 111K,116.211768,0.1065599566767107
C,111N,B(long) 111N,0.106559957,20.061623176440904
B,111J,c(long) 111J,20.061634,0.0038113334533441123

Answer 1

import csv

HEADER = "Type,Id,TypeId,CalcValues,ID"
final_results = ['0.1065599566767107', '20.061623176440904', '0.0038113334533441123']

with open("test.csv") as inputs, open("tmp.csv", "wb") as outputs:
    reader = csv.reader(inputs, delimiter=",")
    writer = csv.writer(outputs, delimiter=",")

    reader.next()  # ignore header line
    writer.writerow(HEADER.split(","))  

    for row in reader:
        writer.writerow(row + [final_results.pop(0)])

我将头字段存储到HEADER，并切换final_results的第二和第三个元素，使用pop(0)移除并返回{}的第一个元素

输出：

^{pr2}$

Answer 2

第一个方法肯定会失败：您不想添加新行，而是要添加新列。所以回到第二种方法：

您可以插入标题，但随后将循环遍历每行的结果，而您需要对它们进行迭代。在

为此，我从final_results列表（使用__iter__()）创建一个迭代器，然后调用it.next并追加到每一行（不需要在末尾插入，只需追加）

我删除了all大列表，因为1）你可以一次写一行，节省内存，2）all是一个预定义的函数。避免将其用作变量。在

final_results = ['0.1065599566767107', '0.0038113334533441123', '20.061623176440904']

# Method2
with open("test.csv", 'rb') as input, open('temp.csv', 'wb') as output:
    reader = csv.reader(input, delimiter = ',')
    writer = csv.writer(output, delimiter = ',')


    row = next(reader)  # read title line
    row.append("Results")
    writer.writerow(row)  # write enhanced title line

    it = final_results.__iter__()  # create an iterator on the result

    for row in reader:
        if row:  # avoid empty lines that usually lurk undetected at the end of the files
            try:
                row.append(next(it))  # add a result to current row
            except StopIteration:
                row.append("N/A")     # not enough results: pad with N/A
            writer.writerow(row)

结果：

^{pr2}$

注意：如果我们在final_results变量中包含"Results"，我们甚至不需要以不同的方式处理第一行。在

注2：值似乎错误：final_results的顺序与预期输出的顺序不同。并且Result列已经变成了ID，但这很容易纠正。在