我有一个numpy数组,我想从该数组创建一个字典。
更具体地说,我想要一个字典,它有对应于行的键,所以键1应该是行1的和。
s1是我的数组,我知道如何得到行的和,但在这里我是行。
我正在考虑创建一个循环,在这个循环中,我可以计算行的和,然后将其添加到字典中,但我对编程还不是很熟悉,所以我不知道该如何做,或者如果可能的话。
有人有什么建议吗?
编辑
我用range函数创建了键值。
然后压缩键和数组。

mydict = dict(zip(keys, s1))

最佳答案

我会在精神上做一些类似于你的事情,做两个小改动。
首先,我们可以使用dict(zip(keys, s1)),其次,我们可以调用enumerates的sum方法。示例:

>>> arr = np.arange(9).reshape(3,3)
>>> arr
array([[0, 1, 2],
       [3, 4, 5],
       [6, 7, 8]])
>>> arr.sum(axis=1)
array([ 3, 12, 21])
>>> dict(enumerate(arr.sum(axis=1)))
{0: 3, 1: 12, 2: 21}

10-07 20:00
查看更多