python - examples - plt annotate

Conteo rectangular de Python Matplotlib

Question

Tengo una serie de valores (x, y) que quiero trazar un histograma en 2d usando matplotlib de python. Usando hexbin, obtengo algo como esto: texto alternativo http://img121.imageshack.us/img121/1339/hexbin.png Pero estoy buscando algo como esto: texto alternativo http://img17.imageshack.us/ img17 / 1927 / recthist.png Código de ejemplo:

from matplotlib import pyplot as plt
import random
foo = lambda : random.gauss(0.0,1.0)
x = [foo() for i in xrange(5000)]
y = [foo() for i in xrange(5000)]
pairs = zip(x,y)
#using hexbin I supply the x,y series and it does the binning for me
hexfig = plt.figure()
hexplt = hexfig.add_subplot(1,1,1)
hexplt.hexbin(x, y, gridsize = 20)
#to use imshow I have to bin the data myself
def histBin(pairsData,xbins,ybins=None):
if (ybins == None): ybins = xbins
xdata, ydata = zip(*pairsData)
xmin,xmax = min(xdata),max(xdata)
xwidth = xmax-xmin
ymin,ymax = min(ydata),max(ydata)
ywidth = ymax-ymin
def xbin(xval):
xbin = int(xbins*(xval-xmin)/xwidth)
return max(min(xbin,xbins-1),0)
def ybin(yval):
ybin = int(ybins*(yval-ymin)/ywidth)
return max(min(ybin,ybins-1),0)
hist = [[0 for x in xrange(xbins)] for y in xrange(ybins)]
for x,y in pairsData:
hist[ybin(y)][xbin(x)] += 1
extent = (xmin,xmax,ymin,ymax)
return hist,extent
#plot using imshow
imdata,extent = histBin(pairs,20)
imfig = plt.figure()
implt = imfig.add_subplot(1,1,1)
implt.imshow(imdata,extent = extent, interpolation = ''nearest'')
plt.draw()
plt.show()

Parece que ya debería haber una forma de hacerlo sin escribir mi propio método de "agrupamiento" y usar imshow.

Answer 1

¿ matplotlib.pyplot.hist es lo que estás buscando?

>>> help(matplotlib.pyplot.hist)
Help on function hist in module matplotlib.pyplot:
hist(x, bins=10, range=None, normed=False, weights=None, cumulative=False, botto
m=None, histtype=''bar'', align=''mid'', orientation=''vertical'', rwidth=None, log=Fa
lse, hold=None, **kwargs)
call signature::
hist(x, bins=10, range=None, normed=False, cumulative=False,
bottom=None, histtype=''bar'', align=''mid'',
orientation=''vertical'', rwidth=None, log=False, **kwargs)
Compute and draw the histogram of *x*. The return value is a
tuple (*n*, *bins*, *patches*) or ([*n0*, *n1*, ...], *bins*,
[*patches0*, *patches1*,...]) if the input contains multiple
data.

Answer 2

Numpy tiene una función llamada histogram2d , cuya docstring también le muestra cómo visualizarla usando Matplotlib. Agregue interpolation=nearest a la llamada a imshow para deshabilitar la interpolación.

Answer 3

Acabo de enviar una solicitud de extracción para este https://github.com/matplotlib/matplotlib/pull/805 . Con suerte, será aceptado.

Answer 4

Me doy cuenta de que hay un parche enviado a matplotlib, pero adopté el código del otro ejemplo para acomodar algunas necesidades que tenía.

ahora el histograma se traza desde la esquina inferior izquierda, como en matemáticas convencionales (no computacionales)

también, se ignoran los valores fuera del rango de binning y yo uso una matriz nudy 2d para la matriz bidimensional

Cambié la entrada de datos de pares a dos matrices 1D ya que así es como se suministran los datos para dispersar (x, y) y funciones similares

def histBin(x,y,x_range=(0.0,1.0),y_range=(0.0,1.0),xbins=10,ybins=None):
""" Helper function to do 2D histogram binning
x, y are lists / 2D arrays
x_range and yrange define the range of the plot similar to the hist(range=...)
xbins,ybins are the number of bins within this range.
"""
pairsData = zip(x,y)
if (ybins == None):
ybins = xbins
xdata, ydata = zip(*pairsData)
xmin,xmax = x_range
xmin = float(xmin)
xmax = float(xmax)
xwidth = xmax-xmin
ymin,ymax = y_range
ymin = float(ymin)
ymax = float(ymax)
ywidth = ymax-ymin
def xbin(xval):
return floor(xbins*(xval-xmin)/xwidth) if xmin <= xval < xmax else xbins-1 if xval ==xmax else None
def ybin(yval):
return floor(ybins*(yval-ymin)/ywidth) if ymin <= yval < ymax else ybins-1 if yval ==ymax else None
hist = numpy.zeros((xbins,ybins))
for x,y in pairsData:
i_x,i_y = xbin(x),ybin(ymax-y)
if i_x is not None and i_y is not None:
hist[i_y,i_x] += 1
extent = (xmin,xmax,ymin,ymax)
return hist,extent

Answer 5

Usa xlim y ylim para establecer los límites de la trama. xlim(-3, 3) y ylim(-3, 3) deberían hacerlo.