python - 如何在 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 中，数组 x和 y形状 (256, 1)和 (256) broadcast到一个共同的形状(256, 256) .