How to create a legend for 3D bar in matplotlib?

by

You need to use a proxy artist where legends are not supported.

This code:

from mpl_toolkits.mplot3d import Axes3D
import matplotlib.pyplot as plt
import numpy as np

fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
x, y = np.random.rand(2, 100) * 4
hist, xedges, yedges = np.histogram2d(x, y, bins=4)

elements = (len(xedges) - 1) * (len(yedges) - 1)
xpos, ypos = np.meshgrid(xedges[:-1]+0.25, yedges[:-1]+0.25)

xpos = xpos.flatten()
ypos = ypos.flatten()
zpos = np.zeros(elements)
dx = 0.5 * np.ones_like(zpos)
dy = dx.copy()
dz = hist.flatten()

ax.bar3d(xpos[:8], ypos[:8], zpos[:8], dx, dy, dz, color="b", zsort="average")
blue_proxy = plt.Rectangle((0, 0), 1, 1, fc="b")
ax.bar3d(xpos[8:], ypos[8:], zpos[8:], dx, dy, dz, color="r", zsort="average")
red_proxy = plt.Rectangle((0, 0), 1, 1, fc="r")
ax.legend([blue_proxy,red_proxy],['cars','bikes'])

plt.show()

produces this:enter image description here

More Related Contents:

  1. How to find the exact intersection of a curve (as np.array) with y==0?
  2. pylab.hist(data, normed=1). Normalization seems to work incorrect
  3. how to do multiple scatter plots with matploatlib
  4. Plotting time in Python with Matplotlib
  5. surface plots in matplotlib
  6. Set Colorbar Range in matplotlib
  7. Histogram Matplotlib
  8. Matplotlib log scale tick label number formatting
  9. matplotlib savefig() plots different from show()
  10. Why does pyplot.contour() require Z to be a 2D array?
  11. How to create a density plot in matplotlib?
  12. matplotlib advanced bar plot
  13. Pythonic way of detecting outliers in one dimensional observation data
  14. Polar contour plot in matplotlib – best (modern) way to do it?
  15. How to fix ‘numpy.ndarray’ object has no attribute ‘get_figure’ when plotting subplots
  16. Creating a Confidence Ellipsis in a scatterplot using matplotlib
  17. Using datetime as ticks in Matplotlib
  18. Plot the x-axis as a date
  19. Axes from plt.subplots() is a “numpy.ndarray” object and has no attribute “plot”
  20. new pythonic style for shared axes square subplots in matplotlib?
  21. How to have actual values in matplotlib Pie Chart displayed
  22. How to install NumPy for python 3.3.5 on Mac OSX 10.9
  23. Matplotlib / python clickable points
  24. Adding a figure created in a function to another figure’s subplot
  25. Save plot to numpy array
  26. How to plot multiple functions on the same figure
  27. How to create a density plot
  28. Move X-Axis label downwards, but not X-Axis Ticks in matplotlib
  29. How to add trendline to a scatter plot
  30. Add density curve on the histogram

Leave a Comment