填写python列表中缺少的数据

Question

我有一个列表字典，每个列表都有不同数量的元素。我想在每个列表的开头添加默认值，使它们的大小相同。

所以，如果我开始......

data = {'data1': [1,2,3], 'data2': [12,23,34,45,44], 'data3': [7,8,9,10], 'data4': [71, 72, 73, 74, 75, 76, 78], 'data5': []}

我想最终：

data1: [0, 0, 0, 0, 1, 2, 3]
data2: [0, 0, 12, 23, 34, 45, 44]
data3: [0, 0, 0, 7, 8, 9, 10]
data4: [71, 72, 73, 74, 75, 76, 78]
data5: [0, 0, 0, 0, 0, 0, 0]

我提出了以下代码：

maxlen = 0
for d in data:
    if maxlen < len(data[d]):
        maxlen = len(data[d])

for d in data:
    if len(data[d]) < maxlen:
        data[d] = [0] * (maxlen - len(data[d])) + data[d]

这样可行，但还有更好的方法吗？

Answer 1

你可以找出哪个列表最长，然后在dict理解中用零填充其他每个列表。

>>> longest = max(len(i) for i in data.values())
>>> data = {key: [0]*(longest-len(value)) + value for key,value in data.items()}
>>> data
{'data5': [0, 0, 0, 0, 0, 0, 0],
 'data1': [0, 0, 0, 0, 1, 2, 3],
 'data4': [71, 72, 73, 74, 75, 76, 78],
 'data3': [0, 0, 0, 7, 8, 9, 10],
 'data2': [0, 0, 12, 23, 34, 45, 44]}