python - 对 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)