Вычитание и / или изменение формы массива

Question

Я хотел бы помочь с проблемой. В Python:

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

Я хотел бы знать, есть ли функция, которая позволяет мне вычесть b-a, чтобы получить массив 2x10, полный -1.

Я могу сделать это один с 1D массивами, я просто хотел знать, возможно ли это сделать с 2D массивами.

Спасибо

Answer 1

Добавить новое измерение в a:

b - a[:,None]

, где a[:,None] становится array([[2], [2]]), массивом 2x1, который можно вычесть из массива 2x10 и получить массив 2x10, заполненный -1.

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.]])