python - 使用固定值扩展列表

Question

我有以下列表:

l = [5, 6, 7, 1]

我需要用第一个值(即 5)填充此列表，以便此列表的长度等于 10。

预期结果:

l_extended = [5, 5, 5, 5, 5, 5, 5, 6, 7, 1]

我可以在 for 循环中完成:

fixed_val = l[0]
len_diff = 10 - len(l)
l_extended = []

for n in range(len_diff):
   l_extended.append(fixed_val)

for n in range(len_diff,10):
   l_extended.append(l[n-len_diff])

但是有没有更短的方法呢？

Answer 1

还考虑

a = [1,2,3]
a_extended = [ a[0] ] * ( 10-len(a) ) + a

解释:

a[0] 抓取第一个元素

(10-len(a)) 是我们需要添加的字符数才能使长度达到 10

在Python中，你可以执行[1] * 3得到[1,1,1]，所以:

[a[0]] * (10-len(a)) 根据我们需要的额外元素数量重复第一个元素

在 python 中，你可以执行 [1,2,3] + [4,5,6] 得到 [1,2,3,4,5,6]，所以:

[a[0]]*(10-len(a)) + a 将额外的元素添加到列表的前面