如何在python中的现有csv文件中添加数组作为列？

Question

我有一系列响应变量，我通过使用进一步的代码进行回归分析。

SalePrice_baseline = reg2.predict(test)

输出看起来像，

array([[ 103612.29783843],
       [  95608.74582384],
       [ 178230.07228516],
       ..., 
       [ 172559.98073767],
       [ 121881.65675305],
       [ 218179.82985471]])

我将所有输入变量的现有测试csv文件作为test.csv。我想将此数组添加为新列。

我该怎么做？

谢谢。

Answer 1

如果CSV文件的行数与向量相同，那么 你可以读一行作为一个字符串，然后打印它加上一个逗号加上相应的矢量值

with open('a.csv') as inp:
    with open('b.csv', 'w') as out:
        for line, value in zip(inp, vector)
            line = line.rstrip()
            print(line, ',', value, sep='', file=out)

Answer 2

试试这个：

with open('test.csv', 'w') as csv_file:
    writer = csv.writer(csv_file)
    for row in SalePrice_baseline:
        writer.writerow(row)