python - 使用循环自动解包元组并生成变量名称

Question

这个问题已经有答案了:

Dynamically create variable names? [duplicate] (1 个回答)

Dynamically creating a variable (1 个回答)

How do I create variable variables? (17 个回答)

已关闭 3 年前。

我有一个返回 4 个列表的函数。

def foo ( Input1, input 2)
    return list1, list2, list3, list 4

我现在正在使用以下代码:

x0, y0, z0, w0 = foo(ip0, ip4)
x1, y1, z1, w1 = foo(ip3, ip1)
x2, y2, z2, w2 = foo(ip3, ip4)
x3, y3, z3, w3 = foo(ip1, ip2)

我想减少行数，可以使用 for循环。

我尝试了 while 循环:

i=0
while i<3:
    'x'+str(i), 'y'+str(i),'z'+str(i),'w'+str(i)= foo(input[i], input[i])
    i +=1

Python 抛出错误，它认为我正在为字符串赋值。如何使用循环或自动解压元组并生成变量名称？

Answer 1

你可以这样做:

def MyFunction(arg1, arg2):

    list1 = [arg1, arg2]
    list2 = [arg2, arg1]

    return list1, list2


# create empty dictionary:
my_dictionary = {}

# iterate in given range:
for index in range(4):

    # get output of function:
    list1, list2 = MyFunction(1, 2)

    # assign output to dictionary:
    single_dictionary_entry = {
        'x' + str(index) : list1,
        'y' + str(index) : list2,
    }

    # update original dictionary with new entry:
    my_dictionary.update(single_dictionary_entry)

# print result:
print(my_dictionary)