Вычитание строк в массиве без циклов

Как я могу вычесть две строки в python numpy? без использования петель

Например:

      import numpy as np
x = np.array([[2.2, 3.4, 5.2], [1., 5.4, 5.]])

# expected result:
y = [[-1.2, 2., -0.2]]

2 ответа

Вы можете сделать следующее:

      import numpy as np
x = np.array([[2.2, 3.4, 5.2], [1., 5.4, 5.]])
np.diff(x.T)
# array([[-1.2],
#    [ 2. ],
#    [-0.2]])

Попробуйте вычесть с вычитанием:

      import numpy as np
x = np.array([[2.2, 3.4, 5.2], [1., 5.4, 5.]])
print(x[1] - x[0])
Другие вопросы по тегам