import os
import numpy as np
import matplotlib.pyplot as plt

# Generate dummy data
delta = 0.01
x = np.arange(-3.0, 3.0, delta)
y = np.arange(-3.0, 3.0, delta)
X, Y = np.meshgrid(x, y)
Z = X * Y

# Set range and steps for colour bar
c_max = 5
c_min = -5
c_step = 1

# Create colour map
cmap = plt.cm.get_cmap('YlOrRd', (c_max - c_min) / c_step)

fig, ax = plt.subplots(1, 1, figsize=(6, 4))

# Plot image
im = ax.imshow(Z,
               cmap=cmap,
               interpolation='bilinear',
               vmax=c_max,
               vmin=c_min)

# Add colour bar
plt.colorbar(im, ticks=np.arange(c_min, c_max + 1, c_step))

# Hide spines and ticks
ax.set_axis_off()

# Export figure
png_name = os.path.join(
    'png',
    os.path.basename(__file__).replace('.py', '.png').replace('_', '-'))
plt.savefig(png_name, bbox_inches='tight', dpi=100)