2016-11-29 89 views
-1

我有一個響應變量數組,我通過使用進一步的代碼使用迴歸分析。如何將數組添加爲python中現有csv文件中的列?

SalePrice_baseline = reg2.predict(test) 

輸出的樣子,

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

我所有的輸入變量作爲test.csv的現有測試csv文件。我想添加這個數組作爲一個新的列。

我該怎麼做?

謝謝。

+0

從csv文件讀取所有列,將列添加到內存中的列,將所有列寫入csv文件。 – furas

回答

-1

試試這個:

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

它不會將新列添加到文件中存在的列。 – furas

+0

正確,我的錯誤 –

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) 
+0

您也可以使用內置的csv編寫器,如[其他答案](http://stackoverflow.com/a/40875757/6917168)使用此算法。它會自動轉義值,允許您輕鬆更改分隔符等。 – Lunaweaver

相關問題