gpt4 book ai didi

python - 向 numpy 数组添加索引

转载 作者:太空狗 更新时间:2023-10-30 00:55:22 26 4
gpt4 key购买 nike

我有一个 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 参数的列表来指定每一列的格式:

fmt=['%d', '%1.1f', '%1.1f', '%1.1f']

完整示例:

import numpy as np
prob_rf = [[1, 0.4, 0.4, 0.4],
[2, 0.5, 0.5, 0.5],
[3, 0.6, 0.6, 0.6]]
np.savetxt("foo.csv", prob_rf, delimiter=",", fmt=['%d', '%1.1f', '%1.1f', '%1.1f'])

生成的文件:

1,0.4,0.4,0.4
2,0.5,0.5,0.5
3,0.6,0.6,0.6

关于python - 向 numpy 数组添加索引,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/29208688/

26 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com