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

Question

我有以下numpy数组

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

现在我要对数组的两列进行dd生成，以生成数组的第三列。我正在尝试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)