def test_colorbar_example1():
with cbook.get_sample_data("grace_hopper.png") as fp:
data = np.array(plt.imread(fp))
fig = plt.figure(figsize=(8, 6))
ax = fig.add_subplot("111", aspect="equal")
mappable = ax.imshow(data[..., 0], cmap="viridis")
colorbar = Colorbar(mappable, location="lower left")
colorbar.set_ticks([0.0, 0.5, 1.0])
ax.add_artist(colorbar)
def test_colorbar_example1():
with cbook.get_sample_data('grace_hopper.png') as fp:
data = np.array(plt.imread(fp))
fig = plt.figure()
ax = fig.add_subplot("111", aspect='equal')
mappable = ax.imshow(data[..., 0], cmap='viridis')
colorbar = Colorbar(mappable, location='lower left')
colorbar.set_ticks([0.0, 0.5, 1.0])
ax.add_artist(colorbar)
def test_colorbar_example2():
with cbook.get_sample_data("grace_hopper.png") as fp:
data = np.array(plt.imread(fp))
fig = plt.figure(figsize=(8, 6))
ax = fig.add_subplot("111", aspect="equal")
norm = matplotlib.colors.Normalize(vmin=-1.0, vmax=1.0)
mappable = ax.imshow(data[..., 0], cmap="viridis", norm=norm)
colorbar = Colorbar(mappable, location="lower left")
colorbar.set_ticks([-1.0, 0, 1.0])
ax.add_artist(colorbar)
def create_figure():
fig = plt.figure()
ax = fig.add_subplot("111")
data = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
mappable = ax.imshow(data)
colorbar = Colorbar(mappable)
ax.add_artist(colorbar)
return fig, ax, colorbar
def add_colorbar(ax, im, loc='center left', label=None):
from matplotlib_colorbar.colorbar import Colorbar
cbar = Colorbar(im, location=loc, border_pad=0.5)
if label is not None:
cbar.set_label(label)
cbar.set_alpha(1)
#cbar.draw_all()
ax.add_artist(cbar)
"""
Draw the example used in the README file.
"""
# Standard library modules.
# Third party modules.
import numpy as np
import matplotlib.pyplot as plt
import matplotlib.cbook as cbook
# Local modules.
from matplotlib_colorbar.colorbar import Colorbar
# Globals and constants variables.
plt.figure()
data = np.array(plt.imread(cbook.get_sample_data('grace_hopper.png')))
mappable = plt.imshow(data[..., 0], cmap='viridis')
colorbar = Colorbar(mappable, location='lower left')
colorbar.set_ticks([0.0, 0.5, 1.0])
plt.gca().add_artist(colorbar)
plt.savefig('example1.png', bbox_inches='tight')
