在Python中的特定列中广播2D数组

如何解决在Python中的特定列中广播2D数组

我有一个像这样的数组：

A = np.array([[ 1,2,3,4,5],[ 6,7,8,9,10],[11,12,13,14,15],[16,17,18,19,20]])

我想做的是在第一列和最后一列的每个值上加1。我想通过使用此向量和适当的向量来理解广播（避免循环），但是我已经尝试过了，但是不起作用。预期结果：

A = np.array([[ 2,6],[ 7,11],[12,16],[17,21]])

解决方法

您可以使用numpy indexing来执行此操作。试试这个：

# 0 is the first and -1 is the last column
A[:,[0,-1]]  = A[:,-1]]+1  

或

A[:,(0,-1)]  = A[:,-1)]+1 

或

A[:,-1]]+=1

或

A[:,-1)]+=1 

两种情况下的输出：

array([[ 2,2,3,4,6],[ 7,7,8,9,11],[12,12,13,14,16],[17,17,18,19,21]])

,

您可以使用向量[1,1]，而python会为您进行广播。

+
23
-
23
+
4
=
F
1
Q
a
;

如果您想了解广播的工作原理，只需尝试自己广播一次即可。

b = np.array([1,1])
A + b
array([[ 2,21]])

结果相同。