如何从同一个数组中减去两列，然后用 numpy 将值放入它们自己的单列数组中？答案

【问题标题】：How do I subtract two columns from the same array and put the value in their own single column array with numpy?如何从同一个数组中减去两列，然后用 numpy 将值放入它们自己的单列数组中？
【发布时间】：2019-12-30 19:00:12
【问题描述】：

例如，假设我有一个 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 中解决这个问题？

【问题讨论】：

标签： python arrays numpy subtraction

【解决方案1】：

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

【讨论】：

如果我不知道最终数组将有多少行，我是否可以使用 .reshape((-1,1)) 代替？谢谢