如何繪制離散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)