在2D中使用类似十六进制的箱计数箱

时间:2017-07-13 13:58:41

标签: python numpy scikit-learn

我使用numpy's historgram 2d 计算每个中有多少(训练)数据点。对于新点(x,y),我可以查询点与(x,y)在同一个区域中的位置:

enter image description here

像matplotlib图中的“hex”箱子有类似的东西

enter image description here

我可以在哪里填充垃圾箱然后再查询每个垃圾箱中的点数?

1 个答案:

答案 0 :(得分:1)

您可以获取bin数据,但这并不像在矩形网格上执行相同的操作那么简单。原因是hex bin不适合直接的二维索引。函数hexbin()返回PolyCollection,其{bin}位置可通过get_offsets()访问,bin值可通过get_array()访问。所以:

import matplotlib.pyplot as plt

hb = plt.hexbin(...)

bin_xy = hb.get_offsets()
bin_counts = hb.get_array()