どのようにして平面上に任意の色関数をプロットできますか？

Question

Pythonでは、点（x、y）をとり、色を返すpython関数を記述したいとします。その関数を長方形にプロットする最も簡単な方法は何ですか？私はすでにnumpyとmatplotlibを持っていますが、私は他の無料のライブラリをインストールするつもりです。経験から、matplotlibのはそれを痛みのビットを作るように見える、私はより複雑な何かをする必要がありますする必要はありません以下：

from library import plot 

def f(x, y): 
    # etc 

plot(f, xmin=0, xmax=1, ymin=0, ymax=1) 

Answer 1

以下は色機能fマップはxを調整する例だろうとyをRGBアレイに割り当てます。グリッド上に入力座標を入力すると、fと評価され、イメージとして表示されます。

import numpy as np 
import matplotlib.pyplot as plt 

# color function, from input (x,y) in the range between 0 and 1 
# return an RGB tuple 
f = lambda x,y: np.stack((x,np.zeros_like(x),y), axis=2) 

# input grid of values between 0 and 1 
X,Y = np.meshgrid(np.linspace(0,1,51),np.linspace(0,1,51)) 
# plot function: 
plt.imshow(f(X,Y)) 

plt.show()