python - 如何使用 numpy 从同一数组中减去两列并将值放入它们自己的单列数组中？

Question

假设我有一个 3x4 的数组(3 行，4 列)

import numpy as np

data = [[0,5,0,1], [0,5,0,1], [0,5,0,1]]

data = np.array(data)

print(data)

[[0 5 0 1]
 [0 5 0 1]
 [0 5 0 1]]

我想从第 2 列中减去第 4 列，并将值放在自己的命名 3x1 数组中，如下所示

print(subtraction)

[[4]
 [4]
 [4]]

我该如何在 numpy 中解决这个问题？

Answer 1

result = (data[:, 1] - data[:, 3]).reshape((3, 1))