如何在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

And The output I get is:

0. [add, 6]
1. [subs, -2]
2. [add, 11]
3. [subs, 3]
4. [add, 12]
5. [subs, 4]

But I want the output to be like:

0. [add, 6, 11, 12]
1. [subs, -2, 3, 4]

Please help me

Answer 1

Why don't you do a one-liner comprehension without even having to define a calc() function:

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

Then you can print the inner lists:

for l in output:
   print(l)