python - 是否可以进一步优化此数字运算算法？

Question

我正在尝试在 Python 中进行一些数据处理，并且我有一个嵌套循环可以进行一些算术计算。内部循环执行了 20.000 次，因此以下代码需要很长时间:

for foo in foo_list:
    # get bar_list for foo
    for bar in bar_list:
        # do calculations w/ foo & bar

这个循环使用 Numpy 或 Scipy 会更快吗？

Answer 1

使用 Numpy:

import numpy as np
foo = np.array(foo_list)[:,None]
bar = np.array(bar_list)[None,:]

然后

foo + bar

或其他操作使用相应的结果创建一个数组 len(foo) * len(bar)。

例子:

>>> foo_list = [10, 20, 30]
>>> bar_list = [4, 5]
>>> foo = np.array(foo_list)[:,None]
>>> bar = np.array(bar_list)[None,:]
>>> 2 * foo + bar

array([[24, 25],
       [44, 45],
       [64, 65]])