Numpy Array to Graph

Question

I have a simple 2D Numpy array consisting of 0s and 1s. Is there a simple way to make a graph that will shade in corresponding coordinates?

For example if my array was [[1,0],[0,1]] The plot would be a 2x2 square with the top left and bottom right shaded in

Simon Gibbons · Accepted Answer · 2015-01-30 19:21:27Z

8

You can use matplotlib to plot a matrix for you.

Use the matshow command with an appropriate colourmap to produce the plot.

For example

import numpy as np
import matplotlib.pyplot as plt

x = np.array([[1,0],[0,1]])
plt.matshow(x, cmap='Blues')
plt.show()

would produce:

matshow example

answered Jan 30, 2015 at 19:09

Simon Gibbons

7,2241 gold badge24 silver badges35 bronze badges

Sign up to request clarification or add additional context in comments.

1 Answer 1