在Python中制作相同大小的列表

Question

我有3个不同长度的列表。

我想用“X”附加较短的列表，并使大小等于最长列表的长度。

A = [10,20,30,40,50] 
B = ["A", "B", "C"] 
C = ["X1", "X2"] 

追加 “X” 后，它应该是这样的：

A = [10,20,30,40,50] 
B = ["A", "B", "C", "X","X"] 
C = ["P1", "P2", "X", "X", "X"] 

我用下面的代码实现它，

for i, a in enumerate(A): 
    if i < len(B): 
     ; 
    else: 
     B.append('X') 

我如何能做到这一点有效在python中？

Answer 1

计算每个列表的最大长度并附加增量。

在Python中，List有一个二进制运算符+将多个列表连接在一起，以及*来拼贴。

A = [10,20,30,40,50] 
B = ["A", "B", "C"] 
C = ["X1", "X2"] 
max_length = max(max(len(A), len(B)), len(C)) 
A += ['X'] * (max_length - len(A)) 
B += ['X'] * (max_length - len(B)) 
C += ['X'] * (max_length - len(C)) 

然后使用容器列表来组织它们，以减少重复代码和更好的扩展性。

A = [10,20,30,40,50] 
B = ["A", "B", "C"] 
C = ["X1", "X2"] 

arrays = [A, B, C] 
max_length = 0 
for array in arrays: 
    max_length = max(max_length, len(array)) 

for array in arrays: 
    array += ['X'] * (max_length - len(array)) 

结果：

print(A) # [10, 20, 30, 40, 50] 
print(B) # ['A', 'B', 'C', 'X', 'X'] 
print(C) # ['X1', 'X2', 'X', 'X', 'X'] 

Answer 2

尝试使用max()来获得最大的长度，然后添加列表B和C。

如果你想与P更换X，您可以使用列表理解[i.replace('X','P') for i in C]获得['P1','P2']：

>>> m=max(len(A),len(B),len(C)) 
>>> B+['X']*(m-len(B)) 
['A', 'B', 'C', 'X', 'X'] 
>>> [i.replace('X','P') for i in C]+['X']*(m-len(C)) 
['P1', 'P2', 'X', 'X', 'X'] 

Answer 3

使用extend方法

B.extend(['X'] * (len(A)-len(B))) 

Answer 4

写入功能，使这个为你

A = [10, 20, 30, 40, 50] 
B = ["A", "B", "C"] 
C = ["X1", "X2"] 


def extend_with_extra_elements(*some_lists): 
    max_some_lists_length = max(map(len, some_lists)) 
    for some_list in some_lists: 
     extra_elements_count = max_some_lists_length - len(some_list) 
     extra_elements = ['X'] * extra_elements_count 
     yield some_list + extra_elements 


A, B, C = extend_with_extra_elements(A, B, C) 

足够高效

Answer 5

python itertools模块有许多漂亮的函数，适用于这种情况。例如：

>>> from itertools import izip_longest, izip 
>>> A = [10, 20, 30, 40, 50] 
>>> B = ["A", "B", "C"] 
>>> C = ["X1", "X2"] 
>>> A, B, C = (list(x) for x in (izip(*izip_longest(A, B, C, fillvalue='X')))) 
>>> A 
[10, 20, 30, 40, 50] 
>>> B 
['A', 'B', 'C', 'X', 'X'] 
>>> C 
['X1', 'X2', 'X', 'X', 'X']