You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
40 lines
844 B
Python
40 lines
844 B
Python
5 years ago
|
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)
|