How to Plot a 2D Matrix in Python with Colorbar? (Like Imagesc in Matlab)

In Matlab I can visualize a matrix data quite easily with

data = rand(10,10); % Createas a 10 x 10 random matrix
imagesc(data);
colorbar;

Now I want to do the same thing in python. I already know how to plot a 2d matrix (numpy array):

from matplotlib.pyplot import imshow
import numpy as np
data = np.random.random((10,10))
imshow(np.asarray(img))

but I don't know how to add a colorbar to it. Any ideas?

2 Answers

import numpy as np
import matplotlib.pyplot as plt

plt.imshow(np.random.random((50,50)))
plt.colorbar()
plt.show()

Another possibility is to use plt.matshow()

import numpy as np
import matplotlib.pyplot as plt

plt.matshow(np.random.random((50,50)));
plt.colorbar()
plt.show()

Your Answer

By clicking “Post Your Answer”, you agree to our terms of service, privacy policy and cookie policy

Maya Lin-Takahashi

Maya Lin-Takahashi

Consumer Tech & Gadget Reviewer

Maya is a hardware enthusiast who tests and reviews smart home devices, smartphones, wearables, and audio gear. She focuses on practical consumer value and build quality.

Share this article
Twitter Facebook Pinterest