将值分配给Numpy数组中的不同索引位置

将值分配给Numpy数组中的不同索引位置,第1张

将值分配给Numpy数组中的不同索引位置

您可以使用元组索引:

>>> import numpy as np>>> a = np.zeros((4,2))>>> vals = [4,3,2,1]>>> pos = [(0,0),(1,1),(2,0),(3,1)]>>> rows, cols = zip(*pos)>>> a[rows, cols] = vals>>> aarray([[ 4.,  0.],       [ 0.,  3.],       [ 2.,  0.],       [ 0.,  1.]])


欢迎分享,转载请注明来源:内存溢出

原文地址: http://outofmemory.cn/zaji/5647438.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-12-16
下一篇 2022-12-16

发表评论

登录后才能评论

评论列表(0条)

保存