python - 数组减法和/或 reshape

Question

我需要一些帮助来解决问题。在 Python 中:

a=array([2,2])
b=ones((2,10))

我想知道是否有一个函数可以让我减去 b-a 得到一个 2x10 的数组，其中全是 -1。

我可以用一维数组来做，我只是想知道是否可以用二维数组来做。

谢谢

Answer 1

向a添加一个新维度:

b - a[:,None]

其中 a[:,None] 变为 array([[2], [2]])，一个 2x1 数组，您可以从 2x10 数组中减去它得到一个充满 -1 的 2x10 数组。