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 = 8
c_min = -8
c_step = 2

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

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

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

# 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)