如何从同一数组中减去两列并使用 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]]

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

print(subtraction)
[[4]
[4]
[4]]

我将如何在 Numpy 中做到这一点？

Answer 1

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