我有一个大小 < N的列表,我想用一个值将它填充到大小 N。
当然,我可以使用类似以下的东西,但我觉得应该有一些我错过的东西:
>>> N = 5
>>> a = [1]
>>> map(lambda x, y: y if x is None else x, a, ['']*N)
[1, '', '', '', '']
我有一个大小 < 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, '')
我认为这种方法更加直观和 Pythonic。
a = (a + N * [''])[:N]
没有内置功能。但是您可以为您的任务(或任何东西:p)编写内置插件。
(根据 itertoolpadnone
和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, '', '', '', '']
more-itertools
是一个包含padded
此类问题的特殊工具的库:
import more_itertools as mit
list(mit.padded(a, "", N))
# [1, '', '', '', '']
或者,more_itertools
还实现了 Python itertools 配方,包括padnone
@kennytmtake
提到的,因此不必重新实现它们:
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, '', '', '', '']
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)
extra_length = desired_length - len(l)
l.extend(value for _ in range(extra_length))
这避免了任何额外的分配,这与任何依赖于创建和附加列表的解决方案不同[value] * extra_length
。“extend”方法首先调用迭代器,并在从迭代器中填充之前__length_hint__
将分配扩展这么多。l
离开 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, '', '', '', '']
使用迭代器并利用默认参数next
:
i = iter(a)
a = [next(i, '') for _ in range(N)]
无论哪种方式,我们都想生产N
物品。因此for _ in range(N)
. 那么元素应该是尽可能多a
的,剩下的就可以了''
。使用迭代器,a
我们抓取所有可能的元素,当我们得到 时StopIteration
,将返回默认值''
,即.
添加到现有列表中np.repeat
:
import numpy as np
a + list(np.repeat([''], (N - len(a))))
在元素列表之前添加填充
a[:0] += [''] * (N - len(a))
在元素列表之后添加填充
a += [''] * (N - len(a))
一种用空元素填充列表的 Pythonic 方法是使用列表推导。
my_list = [1,2]
desired_len = 3
# Ensure that the length of my list is 3 elements
[my_list.extend(['']) for _ in range(desired_len - len(my_list))]
[my_list.pop() for _ in range(len(my_list)-desired_len )]