如何计算二维numpy数组的单列总和

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

您可以使用numpy索引

对于第一列： np.sum(np_city[:, 0])

对于第二列： np.sum(np_city[:, 1])