python - 数组减法和/或重塑

Question

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

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

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

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

谢谢

Answer 1

将新维度添加到a：

b - a[:,None]

wherea[:,None]变成array([[2], [2]])一个 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.]])