使用numpy数组避免for循环组合

Question

一定有一种更像Python的方式：

r = np.arange(100)
results = []

for i in r:
    for j in r:
        for k in r:
            for l in r:

                #Here f is some predefined function
                if f(i,j,k,l) < 5.0:
                     results.append(f(i,j,k,l))

我确信使用数组可以简化这一点，但我不确定如何简化。谢谢！在

Answer 1

使用itertools笛卡尔积：

import itertools
r = np.arange(100)
results = []
for (i,j,k,l) in itertools.product(r,repeat=4):
    if f(i,j,k,l) < 5.0:
         results.append(f(i,j,k,l))

或者更简洁的方式，使用列表理解：

^{pr2}$