Một số tích hợp sẵn để đệm danh sách trong python

Tôi có một danh sách có kích thước < N và tôi muốn đệm nó lên kích thước N với một giá trị.

Chắc chắn, tôi có thể sử dụng một cái gì đó như sau, nhưng tôi cảm thấy rằng có một cái gì đó tôi đã bỏ lỡ:

>>> N = 5
>>> a = [1]
>>> map(lambda x, y: y if x is None else x, a, ['']*N)
[1, '', '', '', '']

a += [''] * (N - len(a))

hoặc nếu bạn không muốn thay đổi atại chỗ

new_a = a + [''] * (N - len(a))

bạn luôn có thể tạo một lớp con của danh sách và gọi phương thức bất cứ điều gì bạn muốn

class MyList(list):
    def ljust(self, n, fillvalue=''):
        return self + [fillvalue] * (n - len(self))

a = MyList(['1'])
b = a.ljust(5, '')

Tôi nghĩ rằng cách tiếp cận này là trực quan hơn và mang tính biểu tượng.

a = (a + N * [''])[:N]

Không có chức năng tích hợp cho việc này. Nhưng bạn có thể soạn các cài sẵn cho tác vụ của mình (hoặc bất cứ thứ gì: p).

(Sửa đổi từ itertool's padnonevà takecông thức nấu ăn)

from itertools import chain, repeat, islice

def pad_infinite(iterable, padding=None):
   return chain(iterable, repeat(padding))

def pad(iterable, size, padding=None):
   return islice(pad_infinite(iterable, padding), size)

Sử dụng:

>>> list(pad([1,2,3], 7, ''))
[1, 2, 3, '', '', '', '']

câu trả lời của gnibbler đẹp hơn, nhưng nếu bạn cần một bản nội trang, bạn có thể sử dụng itertools.izip_longest( zip_longesttrong Py3k):

itertools.izip_longest( xrange( N ), list )

điều này sẽ trả về danh sách các bộ giá trị được ( i, list[ i ] )điền vào Không có. Nếu bạn cần loại bỏ bộ đếm, hãy làm điều gì đó như:

map( itertools.itemgetter( 1 ), itertools.izip_longest( xrange( N ), list ) )

Bạn cũng có thể sử dụng một trình tạo đơn giản mà không cần bất kỳ bản dựng nào. Nhưng tôi sẽ không đệm danh sách, mà để logic ứng dụng xử lý một danh sách trống.

Nhưng dù sao, trình vòng lặp không có bản dựng

def pad(iterable, padding='.', length=7):
    '''
    >>> iterable = [1,2,3]
    >>> list(pad(iterable))
    [1, 2, 3, '.', '.', '.', '.']
    '''
    for count, i in enumerate(iterable):
        yield i
    while count < length - 1:
        count += 1
        yield padding

if __name__ == '__main__':
    import doctest
    doctest.testmod()

Nếu bạn muốn pad bằng None thay vì '', map () thực hiện công việc:

>>> map(None,[1,2,3],xrange(7))

[(1, 0), (2, 1), (3, 2), (None, 3), (None, 4), (None, 5), (None, 6)]

>>> zip(*map(None,[1,2,3],xrange(7)))[0]

(1, 2, 3, None, None, None, None)

more-itertoolslà một thư viện bao gồm một paddedcông cụ đặc biệt cho loại vấn đề này:

import more_itertools as mit

list(mit.padded(a, "", N))
# [1, '', '', '', '']

Ngoài ra, more_itertoolscũng triển khai các công thức của Python itertools bao gồm padnonevà takenhư được đề cập bởi @kennytm, vì vậy chúng không cần phải thực hiện lại:

list(mit.take(N, mit.padnone(a)))
# [1, None, None, None, None]

Nếu bạn muốn thay thế phần Noneđệm mặc định , hãy sử dụng cách hiểu danh sách:

["" if i is None else i for i in mit.take(N, mit.padnone(a))]
# [1, '', '', '', '']

Để đi ra khỏi kennytm:

def pad(l, size, padding):
    return l + [padding] * abs((len(l)-size))

>>> l = [1,2,3]
>>> pad(l, 7, 0)
[1, 2, 3, 0, 0, 0, 0]

bạn có thể sử dụng * toán tử giải nén có thể lặp lại :

N = 5
a = [1]

pad_value = ''
pad_size = N - len(a)

final_list = [*a, *[pad_value] * pad_size]
print(final_list)

đầu ra:

[1, '', '', '', '']

extra_length = desired_length - len(l)
l.extend(value for _ in range(extra_length))

Điều này tránh bất kỳ phân bổ bổ sung nào, không giống như bất kỳ giải pháp nào phụ thuộc vào việc tạo và bổ sung danh sách [value] * extra_length. Phương thức "mở rộng" lần đầu tiên gọi __length_hint__đến trình vòng lặp và mở rộng phân bổ cho lnhiều thứ đó trước khi điền nó vào từ trình vòng lặp.