如何使用 Python 输出嵌套循环的结果 [关闭]

Question

我有一个带有 3 个参数的函数。我想计算该函数并输出每个参数组合的结果。

我已经定义了我的函数F(A,B,C)，现在我们可以假设它的输出是一个单一的值f(A,B,C)。

然后我运行一个嵌套循环：

for A in range(1,2):
    for B in range(4,5):
        for C in range(7,8,9):
            F(A,B,C)

但它会覆盖。我想要得到的是这样的数据框：

A   B   C   Output
1   4   7   f(1,4,7)
1   4   8   f(1,4,8)
1   4   9   f(1,4,9)
1   5   7   f(1,5,7)
1   5   8   f(1,5,8)
1   5   9   f(1,5,9)
2   4   7   f(2,4,7)
2   4   8   f(2,4,8)
2   4   9   f(2,4,9)
2   5   7   f(2,5,7)
2   5   8   f(2,5,8)
2   5   9   f(2,5,9)

Answer 1

range(x,y) 不包含值 y

例如range(1,3) 包含数字[1,2] NOT [1,2,3]

for A in range(1,3):
    for B in range(4,6):
        for C in range(7,10):
            print(A,B,C)

输出：

1 4 7
1 4 8
1 4 9
1 5 7
1 5 8
1 5 9
2 4 7
2 4 8
2 4 9
2 5 7
2 5 8
2 5 9

这是你的范围在做什么

• range(1,2) --> [1]
• range(4,5) --> [4]
• range(7,8,9) --> [7] （这个更糟糕，因为第三个参数是步长值）

如您所见，您只是创建了 3 个范围，其中只有一个项目，因此当您循环遍历它时，它永远不会像您想要的那样重复，因为每个范围只有一个值

这就是为什么您的程序输出 (1,4,7) 然后在您将 F 更改为打印时终止的原因。在我的示例中，我将每个范围的第二个参数增加了 1，以获得您想要的结果。

将其设为数据框：

listData = []
for A in range(1, 3):
    for B in range(4, 6):
        for C in range(7, 10):
            listData.append([A,B,C, A+B+C]) #I just used A+B+C instead of the func f

df = pandas.DataFrame(columns = ["A", "B", "C", "Output"], data = listData)
print(df)

Answer 2

我的建议是使用itertools.product

import itertools
import pandas
# Assumes you know the inputs
A = [1, 2]
B = [4, 5]
C = [7, 8, 9]
# args = list(itertools.product(A, B, C))
args = [list(x) for x in itertools.product(A, B, C)]

def my_func(arg1, arg2, arg3):
    return arg1 + arg2 + arg3

如您所见，上面的代码使您不必使用嵌套循环。

您现在可以执行以下操作：

for arg in args:
    arg.append(my_func(*arg))
df = pandas.DataFrame(args, columns=['A', 'B', 'C', 'SUM'])