如何连接两个不同长度的列表以填充缺少的值

Question

我正在尝试将一个列表中的元素添加到另一个列表中，该列表包含部分元素，但缺少一些元素

days_m = ['1','2','3']
dates = ['4-','10-','10-20','10-4','9-']
start_end = []

for f,b in itertools.zip_longest(dates, days_m, fillvalue = 'NA'):
    if len(f) < 4:
        date = f+b
        start_end.append(date)
    elif len(f) >= 4:
        start_end.append(date)

我希望输出结果是

['4-1', '10-2', '10-20', '10-4', '9-3']

但实际的产出是

['4-1', '10-2', '10-2', '10-2', '9-NA']

Answer 1

首先，您不能在这里使用zip_longest。因为您的意思是只有在长度小于4的情况下才从第二个列表中选取一个值。zip_longest一直都是这样做的。

那么，您将不会在第二个分支中更新date。

我建议在days_m上使用iter，并使用next手动迭代它，默认值为"NA"。同时删除双倍长度测试，因为它是多余的。

并始终使用f的更新版本。只需在需要时连接到它，否则请按原样使用。

days_m = iter(['1','2','3'])
dates = ['4-','10-','10-20','10-4','9-']
start_end = []

for f in dates:
    if len(f) < 4:
        f += next(days_m,"NA")
    start_end.append(f)

print(start_end)

结果：

['4-1', '10-2', '10-20', '10-4', '9-3']