python - 使用函数给定的值初始化 numpy 数组的最快方法

Question

我主要对 ((d1,d2)) numpy 数组(矩阵)感兴趣，但这个问题对于具有更多轴的数组很有意义。我有函数 f(i,j)，我想通过这个函数的一些操作来初始化一个数组

A=np.empty((d1,d2))
for i in range(d1):
    for j in range(d2):
        A[i,j]=f(i,j)

这是可读的并且有效，但我想知道是否有更快的方法，因为我的数组 A 将非常大，我必须优化这一点。

Answer 1

一种方法是使用 np.fromfunction .您的代码可以替换为以下行:

np.fromfunction(f, shape=(d1, d2))

This is implemented在 NumPy 函数方面，对于更大的数组，应该比 Python for 循环快很多。