numpy无法将值分配给列的一部分

Question

当我想要将矢量分配给矩阵中列的一部分时，它失败。这里是一个小例子：

>>> a = np.random.randn(3,4) 
>>> a 
array([[ 0.12263942, -0.21972043, -0.97263544, 0.10609172], 
     [ 0.27012565, 1.0006549 , 0.80596588, -0.2954019 ], 
     [-0.28402912, -2.62405207, -0.52560166, 0.07325703]]) 
>>> x 
array([-0.33707461, 0.66196137]) 
>>> a[np.array([0,2])][:,2] = x 
>>> a[np.array([0,2])][:,2] 
array([-0.97263544, -0.52560166]) 

但这个工程：

>>> a = np.random.randn(3,4) 
>>> a 
array([[ 0.45379867, 0.38782073, -0.01432557, -1.03457707], 
     [ 1.27650459, 0.4480737 , -1.31625142, 2.00037792], 
     [ 0.77420762, 1.20131941, -0.98937721, -0.04921255]]) 
>>> x2 = np.random.randn(3) 
>>> x2 
array([-1.82217731, -0.72454628, -0.41954297]) 
>>> a[:,2] = x2 
>>> a[:,2] 
array([-1.82217731, -0.72454628, -0.41954297]) 

为什么第一个例子不能工作？我错过了什么吗？

Answer 1

当您在数组上使用[]时，它将返回您正在引用的数组片的副本，或者准备好分配给该特定片。

所以当你做a[slc1][slc2] = stuff，a[slc1]现在的a副本，称之为a_使a[slc1][slc2]相同a_[slc2]。

然后，您将该副本分配给a[slc1][slc2] = stuff与a_[slc2] = stuff相同。这很好，但你已经分配给一个只是临时创建的变量。

而是做同样的作业

a[slc1, slc2] = stuff 

内，或者在你的情况分配

a[[0, 2], 2] = x 

请与(a[[0, 2], 2] == x).all()将返回True