Деякі вбудовані для вкладки списку в python

У мене є список розміру < N, і я хочу змінити його до розміру N зі значенням.

Звичайно, я можу використовувати щось на кшталт наступного, але я відчуваю, що має бути щось, що я пропустив:

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

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

або якщо ви не хочете змінюватися aна місці

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

Ви завжди можете створити підклас списку і викликати метод, що завгодно

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

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

Я думаю, що цей підхід є більш візуальним та пітонічним.

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

Для цього немає вбудованої функції. Але ви можете скласти додатки для свого завдання (або що-небудь: p).

(Змінено з itertool's padnoneі takeрецептів)

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)

Використання:

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

Відповідь gnibbler є приємнішою, але якщо вам потрібен вбудований, ви можете використовувати itertools.izip_longest( zip_longestв Py3k):

itertools.izip_longest( xrange( N ), list )

який поверне список кортежів, ( i, list[ i ] )заповнених None. Якщо вам потрібно позбутися лічильника, зробіть щось на кшталт:

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

Ви також можете використовувати простий генератор без будь-яких вбудованих модулів. Але я б не вкладав цей список, а дозволю логіці програми мати справу з порожнім списком.

У всякому разі, ітератор без збірок

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()

Якщо ви хочете виконати накладку на "None" замість "", map () виконує завдання:

>>> 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-itertoolsце бібліотека, яка включає спеціальний paddedінструмент для подібного роду проблем:

import more_itertools as mit

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

Крім того, more_itertoolsтакож реалізуються рецепти itertools Python, включаючи padnoneта takeяк зазначено у @kennytm, тому їх не потрібно повторно виконувати:

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

Якщо ви хочете замінити Noneпрокладку за замовчуванням , використовуйте розуміння списку:

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

Щоб вийти з 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]

Ви можете скористатися * оператором розпакування :

N = 5
a = [1]

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

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

вихід:

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

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

Це дозволяє уникнути зайвого розподілу на відміну від будь-якого рішення, яке залежить від створення та додавання списку [value] * extra_length. Спосіб "розширити" спочатку викликає __length_hint__ітератор і продовжує виділення на lце набагато, перш ніж заповнити його з ітератора.