Question

给定一组长度不同的数组。是否有一种更简洁的方法来通过用以下代码以外的零填充较短的数组来等长数组的长度：

a = [[1.0, 2.0, 3.0, 4.0],[2.0, 3.0, 1.0],[5.0, 5.0, 5.0, 5.0],[1.0, 1.0]]


max =0
for x in a:
    if len(x) > max:
        max = len(x)

print max
new = []
for x in a:
    if len(x)<max:
        x.extend([0.0]* (max-len(x)) )
    new.append(x)

print new

Answer 1

您可以使用以下任一项找到a中最大列表的长度：

len(max(a, key=len))

或

max(map(len, a))

并使用列表推导来构建新列表：

>>> a = [[1.0, 2.0, 3.0, 4.0], [2.0, 3.0, 1.0], [5.0, 5.0, 5.0, 5.0], [1.0, 1.0]]
>>> m = len(max(a, key=len))
>>> new = [x + [0]*(m - len(x)) for x in a]
>>> new
[[1.0, 2.0, 3.0, 4.0], [2.0, 3.0, 1.0, 0], [5.0, 5.0, 5.0, 5.0], [1.0, 1.0, 0, 0]]

Answer 2

In: b = [i+[0.]*(max(map(len,a))-len(i)) for i in a]

In: b
Out: 
[[1.0, 2.0, 3.0, 4.0],
 [2.0, 3.0, 1.0, 0.0],
 [5.0, 5.0, 5.0, 5.0],
 [1.0, 1.0, 0.0, 0.0]]

在数组数组中等待数组的长度

2 个答案: