如何計算二維 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])