如何使用列表作为索引将较小的数组分散到较大的数组中?

2024-04-29 16:14:59 发布

您现在位置:Python中文网/ 问答频道 /正文

我需要帮助才能在Python中获得以下类型的结果(我在Matlab中使用过):

M = numpy.zeros((5,5))
m = numpy.array([[1,2,3],[4,5,6],[7,8,9]])
indx = [0, 2, 3]

# in Matlab: M(indx,indx) = M(indx,indx) + m

因此输出为:

[[ 1.  0.  2.  3.  0.]
 [ 0.  0.  0.  0.  0.]
 [ 4.  0.  5.  6.  0.]
 [ 7.  0.  8.  9.  0.]
 [ 0.  0.  0.  0.  0.]]

Tags: innumpy类型zerosarraymatlabindx