假设我有一个3x4的数组(例如3行,4列)
import numpy as np
data = [[0,5,1],[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进行处理?
result = (data[:,1] - data[:,3]).reshape((3,1))