循环中的多个列表中调用功能

Question

我已经在Python中写了一个功能，该功能采用了三个输入变量。现在，我想在每个输入变量的项目列表中调用该功能。因此，函数调用就是这样：

 call_function('cat', 'brown', 2)

列表是：

 category_list = ['cat', 'dog', 'squirrel']
 color_list = ['brown', 'black']
 age = [1,2,3]

现在我想在每个项目上运行函数。因此，它应该返回猫，棕色和1个，然后返回猫，棕色和2个，然后猫，棕色和3，然后猫，黑色和1个，然后猫，黑色和2个，然后猫，黑色和3然后为狗做同样的事情，然后对松鼠做同样的事情。我知道我需要循环调用该功能，但是我尝试并失败了几次。有人可以帮我吗？

Answer 1

使用 itertools ，它为您完成了工作，调用 product 函数真的很容易。

将变量提供给 itertools.product（....））作为参数
然后 list（）响应。

代码

import itertools
category_list = ['cat', 'dog', 'squirrel']
color_list = ['brown', 'black']
age = [1,2,3]

def call_function(category, color, age):
    print(category, color, age)


combinations = list(itertools.product(category_list, color_list, age))
for combintaion in combinations:
    call_function(combintaion[0], combintaion[1], combintaion[2])

输出

cat brown 1
cat brown 2
cat brown 3
cat black 1
cat black 2
cat black 3
dog brown 1
dog brown 2
dog brown 3
dog black 1
dog black 2
dog black 3
squirrel brown 1
squirrel brown 2
squirrel brown 3
squirrel black 1
squirrel black 2
squirrel black 3

证明

测试在这里

iTertools方法的代码

def product(*args, repeat=1):
    # product('ABCD', 'xy') --> Ax Ay Bx By Cx Cy Dx Dy
    # product(range(2), repeat=3) --> 000 001 010 011 100 101 110 111
    pools = [tuple(pool) for pool in args] * repeat
    result = [[]]
    for pool in pools:
        result = [x+[y] for x in result for y in pool]
    for prod in result:
        yield tuple(prod)

Answer 2

这是使用pd.apply的另一种方法，使用itertools.product

import itertools  

category_list = ['cat', 'dog', 'squirrel']
color_list = ['brown', 'black']
age = [1,2,3]

df=pd.DataFrame.from_records(
    list(itertools.product(
        category_list, color_list, age
    )), columns=['category', 'color', 'age'])

df.apply(lambda row: call_function(row['category'], row['color'], row['age']), axis=1)

结果：

cat brown 1
cat brown 2
cat brown 3
cat black 1
cat black 2
cat black 3
dog brown 1
dog brown 2
dog brown 3
dog black 1
dog black 2
dog black 3
squirrel brown 1
squirrel brown 2
squirrel brown 3
squirrel black 1
squirrel black 2
squirrel black 3

如果函数命名为my_function

df.apply(lambda row: MY_FUNCTION(row['category'], row['color'], row['age']), axis=1)

Answer 3

我通过在循环中调用函数来解决它。我确定有一种更高效/优雅的方法可以做到这一点，但是由于我的DF和列表很小，因此效果很好。想为那些正在处理小列表并遇到类似问题的人发布我的解决方案。当然，欢迎改进建议。

for i in category_list: 
    for j in color_list: 
        for k in age: 
            call_function(i,j,k)