我有一个numpy
数组,如下所示:
prob_rf = [[0.4, 0.4, 0.4],
[0.5, 0.5, 0.5],
[0.6, 0.6, 0.6]]
我想为每个内部数组添加一个索引编号,如下所示:
prob_rf = [[1, 0.4, 0.4, 0.4],
[2, 0.5, 0.5, 0.5],
[3, 0.6, 0.6, 0.6]]
然后使用numpy.savetxt
将该数组保存到csv
文件中。
我当前的操作是:
id = [i for i in xrange(1,len(prob)+1)]
prob_rf = np.insert(prob_rf, 0, id, axis=1)
np.savetxt("foo.csv", prob_rf, delimiter=",", fmt='%1.1f')
但是这会输出 。
[[1.0, 0.4, 0.4, 0.4],
[2.0, 0.5, 0.5, 0.5],
[3.0, 0.6, 0.6, 0.6]]
请问如何获得以下输出结果:
[[1, 0.4, 0.4, 0.4],
[2, 0.5, 0.5, 0.5],
[3, 0.6, 0.6, 0.6]]
fmt ='%1.1f'
格式化所有内容? - Carsten[0.4,0.4,0.4]
,否则 np.savetext 会保存为[4.00E-01,4.00E-01,4.00E-01]
,而这不是我想要的。 - VeilEclipse