Агрегировать 3D-массив с использованием массивов индексов зон и времени ⇐ Python

[Anonymous]

Используя небольшой пример ниже, я пытаюсь агрегировать (суммировать) значения в трехмерном массиве dat_arr, используя два других массива для управления группировкой. Первый индекс dat_arr связан со временем. Второй и третий индексы связаны с пространственными (X, Y) положениями. Как я могу суммировать значения в dat_arr, чтобы временное объединение определялось содержимым tim_idx (той же длины, что и первое измерение dat_arr), а пространственное объединение использовало zon_arr > (имеет те же размеры, что и два последних индекса dat_arr)?

Код: Выделить всё

import numpy as np
import matplotlib.pyplot as plt

zon_arr = np.zeros((3,5))
tim_idx = np.array([0,0,1,1,2,2,3,3])

# set up arbitrary zones
zon_arr[1, :3] = 1
zon_arr[1, 3:] = 2

# plt.imshow(zon_arr)
# plt.show()

# generate arbitrary array with data
# first index = time; last 2 indices represent X-Y pts in space
# last two indices must have same dims as zon_arr
np.random.seed(100)
dat_arr = np.random.rand(8, 3, 5)import numpy as np

Итак, результат, который я ожидаю, даст мне сумму значений, содержащихся в dat_arr для каждого уникального значения в tim_idx и zon_arr. Другими словами, я ожидал бы, что выходные данные будут иметь 3 значения (соответствующие 3 зонам) для каждого из 4 уникальных значений времени в tim_idx?

Подробнее здесь: https://stackoverflow.com/questions/793 ... dex-arrays