Jak dodać pasek kolorów dla wykresu hist2d

Question

Cóż, wiem, jak dodać pasek kolorów do figury, gdy utworzę figurkę bezpośrednio przy pomocy matplotlib.pyplot.plt.

from matplotlib.colors import LogNorm 
import matplotlib.pyplot as plt 
import numpy as np 

# normal distribution center at x=0 and y=5 
x = np.random.randn(100000) 
y = np.random.randn(100000) + 5 

# This works 
plt.figure() 
plt.hist2d(x, y, bins=40, norm=LogNorm()) 
plt.colorbar() 

Ale dlaczego następujące nie działa, i co muszę dodać na wezwanie colorbar(..) aby to działało.

fig, ax = plt.subplots() 
ax.hist2d(x, y, bins=40, norm=LogNorm()) 
fig.colorbar() 
# TypeError: colorbar() missing 1 required positional argument: 'mappable' 

fig, ax = plt.subplots() 
ax.hist2d(x, y, bins=40, norm=LogNorm()) 
fig.colorbar(ax) 
# AttributeError: 'AxesSubplot' object has no attribute 'autoscale_None' 

fig, ax = plt.subplots() 
h = ax.hist2d(x, y, bins=40, norm=LogNorm()) 
plt.colorbar(h, ax=ax) 
# AttributeError: 'tuple' object has no attribute 'autoscale_None' 

Answer 1

Już prawie jesteś z trzecią opcją. Musisz przekazać obiekt mappable do colorbar, aby wiedzieć, co to jest colormap i limity, aby nadać pasek koloru. Które mogą być AxesImage lub QuadMesh itp

W przypadku hist2D, krotka zwrócony w swojej h to zawiera mappable, ale także kilka innych rzeczy też.

Z docs:

Powroty: Zwracana jest wartość (liczy, xedges, yedges Image).

Tak, aby colorbar, po prostu potrzebują Image.

naprawić swój kod:

from matplotlib.colors import LogNorm 
import matplotlib.pyplot as plt 
import numpy as np 

# normal distribution center at x=0 and y=5 
x = np.random.randn(100000) 
y = np.random.randn(100000) + 5 

fig, ax = plt.subplots() 
h = ax.hist2d(x, y, bins=40, norm=LogNorm()) 
plt.colorbar(h[3], ax=ax) 

Alternatywnie:

counts, xedges, yedges, im = ax.hist2d(x, y, bins=40, norm=LogNorm()) 
plt.colorbar(im, ax=ax)