如何在 python 列表中进行嵌套循环

Question

operation = ['add','subs']
num_list1 = [[2,4],[7,4],[8,4]]

def calc(op, x,y):
    if op=='add':
        return x+y
    elif op =='subs':
        return x-y

def preview():
    result =[]
    for x,y in num_list1:
        for op in operation:
            result.append([op, calc(op,x,y)])    
    return result


result_preview = pd.DataFrame(preview())
result_preview

我得到的输出是：

0. [添加，6]
1. [潜艇，-2]
2. [添加, 11]
3. [潜艇，3]
4. [添加, 12]
5. [潜艇，4]

但我希望输出如下：

0. [添加、6、11、12]
1. [替补，-2、3、4]

请帮帮我

Answer 1

为什么不用定义 calc() 函数就可以进行单行理解：

output = [[op] + [x + y if op == "add" else x - y for x, y in num_list1] for op in operation]

然后你可以打印内部列表：

for l in output:
   print(l)