我如何从同一数组中减去两列，然后使用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))