Суммируйте два столбца массива 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)