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

【问题标题】：How to add an array as a column in existing csv file in python?如何在python中的现有csv文件中添加一个数组作为列？
【发布时间】：2016-11-29 21:03:35
【问题描述】：

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

SalePrice_baseline = reg2.predict(test)

输出看起来像，

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

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

我该怎么做？

谢谢。

【问题讨论】：

从 csv 文件中读取所有列，将列添加到内存中的列，将所有列写入 csv 文件。

标签： python arrays csv dataframe regression

【解决方案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)

【讨论】：

您也可以使用内置的 csv 写入器，如 other answer 中的此算法。它将自动转义值，允许您轻松更改分隔符等。