如何绘制离散的f（x，y）输出

Question

我有一个用python 3编写的函数f（x，y），它接受整数输入并输出1或0。

我的计划是用正方形网格上的黑白方块来说明这个功能，但我不知道该怎么做。我该怎么办？

Answer 1

使用matplotlib绘制数据网格使用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()

如果你的f更复杂，你必须先将它矢量化：

f = np.vectorize(f)