Python вызывает функцию много раз с разными аргументами

Question

У меня есть функция, которая принимает два аргумента. Я хочу запустить функцию для каждой возможной комбинации моих входов и сохранить каждое возвращаемое значение. Например:

def foo(a, b): 
    return (a + b) 

if __name__ == "__main__": 
    a = np.array([1., 2., 3.]) 
    b = np.array([5., 6.]) 
    f1 = foo(a[0], b[0]) #6 
    f2 = foo(a[0], b[1]) #7 
    f3 = foo(a[1], b[0]) #etc 
    f4 = foo(a[1], b[1]) 
    f5 = foo(a[2], b[0]) 
    f6 = foo(a[2], b[1]) 

Как я могу назвать f1 через f6 более элегантным способом, как цикл? Это не может быть прямой цикл, потому что a и b имеют различное количество элементов.

Answer 1

itertools.product - это путь к путинской манере. Однако, если вы ищете более сырой, базовый способ, вам понадобятся два цикла, которые являются основным свойством перекрестного продукта.

for i in a: 
    for j in b: 
     foo(i, j) 

Edit:

Пример Обычаи itertools.product:

for i, j in itertools.product(a, b): 
    foo(i, j) 

---

for tup in itertools.product(a, b): 
    foo(*tup) 

Я предпочитаю первый, потому что элементы кортежа могут быть использованы в явном виде, если это необходимо.