如何繪製離散F（X，Y）輸出

Question

我有一個函數f（x，y）的寫入在Python 3這需要整數輸入和輸出是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)