python – Colorplot of 2D array matplotlib

python – Colorplot of 2D array matplotlib

Im afraid your posted example is not working, since X and Y arent defined. So instead of pcolormesh lets use imshow:

import numpy as np
import matplotlib.pyplot as plt

H = np.array([[1, 2, 3, 4],
              [5, 6, 7, 8],
              [9, 10, 11, 12],
              [13, 14, 15, 16]])  # added some commas and array creation code

fig = plt.figure(figsize=(6, 3.2))

ax = fig.add_subplot(111)
ax.set_title(colorMap)
plt.imshow(H)
ax.set_aspect(equal)

cax = fig.add_axes([0.12, 0.1, 0.78, 0.8])
cax.get_xaxis().set_visible(False)
cax.get_yaxis().set_visible(False)
cax.patch.set_alpha(0)
cax.set_frame_on(False)
plt.colorbar(orientation=vertical)
plt.show()

Here is the simplest example that has the key lines of code:

import numpy as np 
import matplotlib.pyplot as plt

H = np.array([[1, 2, 3, 4],
          [5, 6, 7, 8],
          [9, 10, 11, 12],
          [13, 14, 15, 16]])

plt.imshow(H, interpolation=none)
plt.show()

enter

python – Colorplot of 2D array matplotlib

Leave a Reply

Your email address will not be published.