How to plot discrete f(x,y) output

Question

I have a function f(x,y) written in python 3 which takes integer input and outputs either 1 or 0.

My plan is to illustrate this function with black and white squares on a square grid, but i have no idea of how to do it. What could i do?

I think you should look into `matplotlib` module [here](http://matplotlib.org/api/pyplot_api.html) — spicypumpkin, Feb 20 '17 at 23:20
This might be helpful: http://stackoverflow.com/questions/10194482/custom-matplotlib-plot-chess-board-like-table-with-colored-cells — Jwad Aziz, Feb 20 '17 at 23:28

koalo · Accepted Answer · 2017-02-21T13:29:22.530

0

To plot a grid of data with matplotlib use imshow:

import numpy as np
import matplotlib.pyplot as plt

def f(x,y):
    return (x+y)%2

x = np.arange(-5, 5, 1)
y = np.arange(-5, 5, 1)
xx, yy = np.meshgrid(x, y, sparse=True)

zz = f(xx,yy)

fig, ax = plt.subplots()
ax.imshow(zz,interpolation='none',cmap='gray')
plt.show()

If your f is more complex, you have to vectorize it first:

f = np.vectorize(f)

edited Feb 21 '17 at 13:29

answered Feb 20 '17 at 23:29

koalo

2,113
20
31

How to plot discrete f(x,y) output

1 Answers1