在matplotlib中使用hexbin获取垃圾桶坐标 [英] get bins coordinates with hexbin in matplotlib
问题描述
我使用matplotlib的方法hexbin
来计算数据的2d直方图.
但是我想获得六边形中心的坐标,以便进一步处理结果.
I use matplotlib's method hexbin
to compute 2d histograms on my data.
But I would like to get the coordinates of the centers of the hexagons in order to further process the results.
我在结果上使用get_array()
方法获得了值,但是我不知道如何获取垃圾箱坐标.
I got the values using get_array()
method on the result, but I cannot figure out how to get the bins coordinates.
我尝试根据给定的箱数和数据范围来计算它们,但是我不知道每个方向上箱的确切数目. gridsize=(10,2)
应该可以解决问题,但似乎不起作用.
I tried to compute them given number of bins and the extent of my data but i don't know the exact number of bins in each direction. gridsize=(10,2)
should do the trick but it does not seem to work.
有什么主意吗?
推荐答案
我认为这可行.
from __future__ import division
import numpy as np
import math
import matplotlib.pyplot as plt
def generate_data(n):
"""Make random, correlated x & y arrays"""
points = np.random.multivariate_normal(mean=(0,0),
cov=[[0.4,9],[9,10]],size=int(n))
return points
if __name__ =='__main__':
color_map = plt.cm.Spectral_r
n = 1e4
points = generate_data(n)
xbnds = np.array([-20.0,20.0])
ybnds = np.array([-20.0,20.0])
extent = [xbnds[0],xbnds[1],ybnds[0],ybnds[1]]
fig=plt.figure(figsize=(10,9))
ax = fig.add_subplot(111)
x, y = points.T
# Set gridsize just to make them visually large
image = plt.hexbin(x,y,cmap=color_map,gridsize=20,extent=extent,mincnt=1,bins='log')
# Note that mincnt=1 adds 1 to each count
counts = image.get_array()
ncnts = np.count_nonzero(np.power(10,counts))
verts = image.get_offsets()
for offc in xrange(verts.shape[0]):
binx,biny = verts[offc][0],verts[offc][1]
if counts[offc]:
plt.plot(binx,biny,'k.',zorder=100)
ax.set_xlim(xbnds)
ax.set_ylim(ybnds)
plt.grid(True)
cb = plt.colorbar(image,spacing='uniform',extend='max')
plt.show()
这篇关于在matplotlib中使用hexbin获取垃圾桶坐标的文章就介绍到这了,希望我们推荐的答案对大家有所帮助,也希望大家多多支持IT屋!