从numpy数组创建字典

用户名

我有一个numpy数组,我想从该数组创建一个字典。

更具体地说,我想要一个字典,该字典具有与该行相对应的键,因此键1应该是第1行的总和。

s1是我的数组,我知道如何获取行的总和,但是这样做numpy.sum(s1[i]),其中i是行。

我当时正在考虑创建一个循环,在该循环中我可以计算行的总和,然后将其添加到字典中,但是我是编程新手,所以我不确定如何执行此操作或是否可能。

有人有什么建议吗?

编辑

我使用range函数创建了键值。然后压缩键和数组。

mydict = dict(zip(keys, s1))
帝斯曼

我会做一些与您精神相似的事情dict(zip(keys, s1)),但有两个小改动。

首先,我们可以使用enumerate,其次,我们可以调用ssum方法ndarray例:

>>> 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}

本文收集自互联网,转载请注明来源。

如有侵权,请联系[email protected] 删除。

编辑于
0

我来说两句

0条评论
登录后参与评论

相关文章