数组减法和/或整形

Question

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

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

我想知道是否有一个函数可以让我减去ba以获得一个充满-1的2x10数组。

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

谢谢

Answer 1

向 a 添加新维度：

b - a[:,None]

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

Answer 2

您想要一个 2x10 的数组，其中充满 -1。

你为什么不这样做：

b = np.ones((2, 10)) * -1

array([[-1., -1., -1., -1., -1., -1., -1., -1., -1., -1.],
       [-1., -1., -1., -1., -1., -1., -1., -1., -1., -1.]])