У меня есть список размеров < 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 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, вы можете использовать itertools.izip_longest (zip_longest в Py3k):
itertools.izip_longest( xrange( N ), list )
который вернет список кортежей ( i, list[ я ] ) заполненный None. Если вам нужно избавиться от счетчика, сделайте что-то вроде:
map( itertools.itemgetter( 1 ), itertools.izip_longest( xrange( N ), list ) )
Вы также можете использовать простой генератор без встроенных вложений. Но я не стал бы заполнять список, но пусть логика приложения имеет дело с пустым списком.
Во всяком случае, итератор без buildins
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()
Если вы хотите набивать "Нет" вместо "", 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 также реализует Python itertools recipes, включая 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, '', '', '', '']
Чтобы выйти из кенности:
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]
extra_length = desired_length - len(l)
l.extend(value for _ in range(extra_length))
Это позволяет избежать какого-либо дополнительного выделения, в отличие от любого решения, которое зависит от создания и добавления списка [value] * extra_length. Метод " __length_hint__ " сначала вызывает __length_hint__ в итераторе и на столько увеличивает распределение для l, прежде чем заполнять его из итератора.
вы можете использовать * оператор повторяющейся распаковки:
N = 5
a = [1]
pad_value = ''
pad_size = N - len(a)
final_list = [*a, *[pad_value] * pad_size]
print(final_list)
выход:
[1, '', '', '', '']