我对numpy不熟悉,我有一个二维对象数组需要分成更小的矩阵并计算每个矩阵中对象数量以制作热图。我按照此线程的答案创建了简单数组的分组和计数,但不知道如何扩展到二维。以下是我目前的代码:
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]
表示法将返回一个二进制值的数组。我无法在任何地方找到关于这个符号的文档,如果有链接将不胜感激。
np.histogram2d
函数? - Syrtis Major