如何合并一个numpy数组和一个文本列并导出到csv

如何合并一个numpy数组和一个文本列并导出到csv

import numpy as npa = np.random.rand(6,4)b = ['test']*6c = np.column_stack([a,b])np.savetxt('/tmp/out', c, delimiter=',', fmt='%s')

写类似

0.70503807191,0.19298150889,0.962915679186,0.655430709887,test0.586655200042,0.379720344068,0.136924270418,0.547277504174,test0.777238053817,0.642467338742,0.709351872598,0.932239808362,test0.386983024375,0.753005132745,0.124107902275,0.472997270033,test0.169711196953,0.735713880779,0.280588048467,0.726851876957,test0.20578446385,0.379406838045,0.640154333103,0.579077700263,test

到

/tmp/out

。

---

按照Paul的建议，如果您有pandas，则可以很容易地形成一个Dataframe，然后调用其to_csv方法：

import numpy as npimport pandas as pda = np.random.rand(6,4)b = np.asarray(['test']*6)df = pd.Dataframe(a)df['b'] = bdf.to_csv('/tmp/out')