Как я могу найти максимум массива NumPy?

Question

У меня есть массив NumPy flowdata этой формы (50, 100, 14).

Данные в основном соответствуют 50 bins, 100 events in each bin, and 14 days of data per event

Теперь я хочу создать максимум каждого дня (для каждого бина и соответствующего события).

Пока я делаю это:

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 array amax имеет возможность выбрать ось. Это позволяет выбрать максимум по одной или нескольким осям (размерам).

Код, который вы указали, похоже, предоставляет максимум для данного бина за все события / дни. Чтобы получить максимум за данный день, попробуйте:

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