我有一个相当简单的数学操作需要在数组上执行。让我写出这个例子:
A = numpy.ndarray((255, 255, 3), dtype=numpy.single)
# ..
for i in range(A.shape[0]):
for j in range(A.shape[1]):
x = simple_func1(i)
y = simple_func2(j)
A[i, j] = (alpha * x * y + beta * x**2 + gamma * y**2, 1, 0)
基本上,(i, j) 和该值的3个值之间存在映射关系(这是为了可视化)。 我想将其整合并进行向量化处理,但我不确定如何处理或是否可以。谢谢。