如何在 numpy/scipy 中实现像素数学

Question

我目前正在阅读一本使用像素数学的书。

问题是我该怎么做：

pmNewComputedImage("Ringing Paraboloid", 256,256,"(sqr(x)+sqr(y)) mod 256")

这将根据公式(sqr(x)+sqr(y)) mod 256 返回一个 256*256 的图像。

这个想法是我无法在 numpy 中做到这一点。我仍然是numpy的新手。我只需要知道如何将行的平方添加到列中，将其修改为 256 并返回一个新数组。

编辑：

在像素数学中：

在 numpy 和 matplotlib 中：

谢谢。

Answer 1

试试这个：

x = np.arange(256).reshape(-1, 1)
y = np.arange(256)
rng_paraboloid = (x*x + y*y) % 256

import matplotlib.pyplot as plt
plt.imshow(rng_paraboloid, interpolation='nearest')

在 numpy-speak 中，数组 x 和 y 的形状 (256, 1) 和 (256) broadcast 到一个共同的形状 (256, 256)。