Вычитание строк в массиве без циклов
Как я могу вычесть две строки в 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])