查找矩陣中每一列的中位數

Question

有沒有辦法找到ndarray中每一列的中位數。 我嘗試了最愚蠢的方法，在每一行上使用雙循環來逐列獲取元素並對其執行 statstics.median 並將其存儲在列表中。

但是，隨着矩陣維數的增加，時間復雜度也會上升。 Python 有沒有更好的方法來解決這個問題？

arr = np.array([[1, 2, 3],[2,3,4],[3,4,5]])
print(arr)
array([[1, 2, 3],
       [2, 3, 4],
       [3, 4, 5]])

預期 output：

2,3,4

Answer 1

跨列使用 numpy.median()。

arr = np.array([[1, 2, 3],[2,3,4],[3,4,5]])
print(arr)
array([[1, 2, 3],
       [2, 3, 4],
       [3, 4, 5]])

np.median(arr,axis=1)