Как поместить 2D-массив в NumPy?

Я новичок в numpy, и у меня есть двумерный массив объектов, который мне нужно скомпоновать в меньшую матрицу, а затем получить количество объектов в каждом бине, чтобы создать тепловую карту. Я последовал за ответомв этой теме для создания бинов и подсчета для простого массива, но я не уверен, как расширить его до 2 измерений. Вот что у меня так далеко:

data_matrix = numpy.ndarray((500,500),dtype=float)
# fill array with values.

bins = numpy.linspace(0,50,50)
digitized = numpy.digitize(data_matrix, bins)

binned_data = numpy.ndarray((50,50))
for i in range(0,len(bins)):
    for j in range(0,len(bins)):
        k = len(data_matrix[digitized == i:digitized == j]) # <-not does not work
        binned_data[i:j] = k

Постскриптум[digitized == i] запись в массиве вернет массив двоичных значений. Я не могу найти документацию по этой записи нигде. Ссылка будет оценена.

Ответы на вопрос(3)

Ваш ответ на вопрос