Как рассчитать сумму одного столбца двумерного массива

Question

import numpy as np

np_city = np.array([ [1.64, 71.78, 56.73],
               [1.37, 63.45, 23.49],
               [2.64, 28.95, 71.89],
               [1.43, 41.88, 88.25],
               [5.37, 23.81, 17.25],
               [2.54, 19.72, 83.57],
               [9.36, 22.55, 41.36],
               [2.49, 57.81, 11.34],
               [6.71, 43.59, 33.21],
               [1.37, 52.87, 23.17],
               [5.63, 79.45, 37.46] ])

Из этого массива я хочу вычислить сумму только первого столбца или только второго столбца или первого и второго столбца. Как я могу это сделать?

Мне нужно решение типа [40.55] или [505.86] или [40.55, 505.86]

В [27]: np.sum(np_city, 0)

Выход [27]: array([ 40.55, 505.86, 487.72])

Требуется один столбец ...

Answer 1

Вы можете использовать цифровую индексацию

Для первого столбца: np.sum(np_city[:, 0])

Для второго столбца: np.sum(np_city[:, 1])