我如何找到一個numpy數組的最大值？

Question

我有這個形狀(50, 100, 14) flowdata的numpy數組flowdata 。

數據基本上對應於50 bins, 100 events in each bin, and 14 days of data per event

現在，我想創建每天的最大值（針對每個bin和對應的事件）。

現在，我正在這樣做：

flowbinmax = []
for b in range(nbins):  #nbins=50
    floweventsmax = []   #Cleared for each bin

    for e in range(nevents):  #nevents=100
        floweventsmax.append(max(flowdata[b][e]))

    flowbinmax.append(floweventsmax)

有沒有更好/更簡單的方法可以做到這一點？ 我不確定我是否做得正確。

Answer 1

numpy數組amax函數具有選擇軸的功能。 這使您可以選擇一個或多個軸（尺寸）上的最大值。

您提供的代碼似乎在所有事件/日期內為給定的bin提供了最大值。 要獲得給定日期的最大值，請嘗試：

daymax = np.amax(flowdata, axis=2)