对 numpy 数组的两列求和并将其添加为数组的第三列

Question

我有以下 numpy 数组

X = np.random.random_integers(100000000,size=(100000000,2))

现在我想添加数组的两列以生成数组的第三列。我正在尝试，X[3] = X[0]+X[1]但它的形状是(2,)。

最终数组示例：

10    5    15
15    6    21

Answer 1

你可以np.concatenate沿着sum最后一个轴。必须将附加轴添加到 的结果中X.sum(1)，因为要连接的所有数组必须具有相同的维数。None这可以通过/来完成np.newaxis：

np.concatenate([X, X.sum(1)[:,None]], -1)