python增加索引_python – 为numpy数组添加索引

我有一个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]]