您可以使用元组索引：
>>> 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.]])
您可以使用元组索引：
>>> 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.]])
名师互学网 版权所有 (c)2021-2022 ICP备案号：晋ICP备2021003244-6号
