我在试图让一个脚本作为一个函数工作时遇到了一点麻烦。
下面的代码基本上压缩了多个值列表。所以我可以继续添加list4
list5
等等
list1 = ['USA', 'France', 'China']
list2 = [30, 39, 28]
list3 = [59.2, 93.1, 922.1]
dictionary = {}
for i in range(len(list1)):
d[list1[i]] = list2[i], list3[i]
print(dictionary)
我正在寻找这样的输出:
{'USA': (30, 59.2), 'France': (39, 93.1), 'China': (28, 922.1)}
现在我想把它变成一个函数,比如:
def zipMultiple(*values, keys):
dictionary = {}
for i in range(len(keys)):
dictionary[keys[i]] = values[i]
return dictionary
myDict = zipMultiple(list2, list3, keys=list1)
我得到这个错误:
dictionary[keys[i]] = values[i] IndexError: tuple index out of range
我知道这可能是由于分配values[i]
,虽然它是一个列表,但我不知道如何确切地概括它说"为每个values
(列表)创建一个元组作为一个参数。
您需要执行转置操作,以便键与值正确对齐:
def zipMultiple(*values, keys):
dictionary = {}
values = list(map(tuple, zip(*values)))
for i in range(len(keys)):
dictionary[keys[i]] = values[i]
return dictionary
或者,您可以将其写成一行代码(Joffan建议进行改进):
def zipMultiple(*values, keys):
return dict(zip(keys, zip(*values)))
在这两种情况下,输出都是:
{'USA': (30, 59.2), 'France': (39, 93.1), 'China': (28, 922.1)}
解释:
如果打印出函数内部的values
(正如最初传入的那样),您将看到:
([30, 39, 28], [59.2, 93.1, 922.1])
第一个列表包含每个键的第一个元素,第二个列表包含每个键的第二个元素,以此类推。
但是,这不是我们想要的。我们希望第一个元素包含第一个键的所有元素,第二个元素包含第二个键的所有元素,以此类推;我们需要的内容如下:
[
(30, 59.2),
(39, 93.1),
(28, 922.1)
]
使得键以与值相同的方式排列。要做到这一点,我们执行上面所示的转置操作。