在Python中创建单项重复n次的列表

时间:2010-08-11 14:01:34

标签: python list-comprehension multiplication

我知道列表理解能做到这一点,但我想知道是否有更短的(更多的Pythonic?)方法。

我想创建一系列不同长度的列表。每个列表将包含相同的元素e,重复n次(其中n =列表的长度)。如何在不执行

的情况下创建列表
[e for number in xrange(n)]
每个列表

8 个答案:

答案 0 :(得分:638)

你也可以写:

[e] * n

你应该注意,如果e是一个空列表,你会得到一个列表,其中n个引用同一个列表,而不是n个独立的空列表。

效果测试

乍一看似乎重复是创建具有n个相同元素的列表的最快方法:

>>> timeit.timeit('itertools.repeat(0, 10)', 'import itertools', number = 1000000)
0.37095273281943264
>>> timeit.timeit('[0] * 10', 'import itertools', number = 1000000)
0.5577236771712819

但等等 - 这不是一个公平的考验......

>>> itertools.repeat(0, 10)
repeat(0, 10)  # Not a list!!!

函数itertools.repeat实际上并不创建列表,只是创建了一个可用于创建列表的对象(如果您愿意)!让我们再试一次,但转换成一个列表:

>>> timeit.timeit('list(itertools.repeat(0, 10))', 'import itertools', number = 1000000)
1.7508119747063233

因此,如果您需要列表,请使用[e] * n。如果您想懒惰地生成元素,请使用repeat

答案 1 :(得分:122)

>>> [5] * 4
[5, 5, 5, 5]

当重复的项目是列表时要小心。该列表不会被克隆:所有元素都将引用相同的列表!

>>> x=[5]
>>> y=[x] * 4
>>> y
[[5], [5], [5], [5]]
>>> y[0][0] = 6
>>> y
[[6], [6], [6], [6]]

答案 2 :(得分:55)

  

在Python中创建单项重复n次的列表

不可变项目

对于不可变项,如None,字符串,元组或frozensets,你可以这样做:

[e] * 4

请注意,这最好只用于列表中的不可变项(字符串,元组,frozensets),因为它们都指向内存中相同位置的相同项。当我必须使用所有字符串的模式构建一个表时,我经常使用它,这样我就不必给出一对一的映射。

schema = ['string'] * len(columns)

可变物品

我已经使用了Python很长一段时间了,我从来没有见过一个用例,我会用一个可变实例来做上面的事情。相反,要获得一个可变的空列表,设置或字典,你应该做这样的事情:

list_of_lists = [[] for _ in columns]

在这种情况下,下划线只是一个一次性的变量名。

如果你只有这个号码,那就是:

list_of_lists = [[] for _ in range(4)]

_并不是特别的,但如果您不打算使用该变量并使用任何其他名称,您的编码环境样式检查器可能会抱怨。


使用带有可变项的不可变方法的注意事项:

注意使用可变对象执行此操作,当您更改其中一个时,它们都会更改,因为它们都是相同的对象:

foo = [[]] *4
foo[0].append('x')

foo现在返回:

[['x'], ['x'], ['x'], ['x']]

但是对于不可变对象,您可以使它工作,因为您更改了引用,而不是对象:

>>> l = [0] * 4
>>> l[0] += 1
>>> l
[1, 0, 0, 0]

>>> l = [frozenset()] * 4
>>> l[0] |= set('abc')
>>> l
[frozenset(['a', 'c', 'b']), frozenset([]), frozenset([]), frozenset([])]

但同样,可变对象对此没有好处,因为就地操作会更改对象,而不是引用:

l = [set()] * 4
>>> l[0] |= set('abc')    
>>> l
[set(['a', 'c', 'b']), set(['a', 'c', 'b']), set(['a', 'c', 'b']), set(['a', 'c', 'b'])]

答案 3 :(得分:23)

Itertools只有这个功能:

import itertools
it = itertools.repeat(e,n)

当然itertools为您提供迭代器而不是列表。 [e] * n为您提供了一个列表,但是,根据您对这些序列的处理方式,itertools变体可以更有效。

答案 4 :(得分:10)

正如其他人所指出的那样,对可变对象使用*运算符会复制引用,因此如果更改了引号,则将其全部更改。如果要创建可变对象的独立实例,则xrange语法是执行此操作的最Pythonic方法。如果您对一个从未使用过的命名变量感到困扰,可以使用匿名下划线变量。

[e for _ in xrange(n)]

答案 5 :(得分:7)

[e] * n

应该有效

答案 6 :(得分:3)

如果您正在寻找简单的重复,例如:

[1, 2, 3, 1, 2, 3, 1, 2, 3]

简单使用:

[1, 2, 3] * 3

但如果您正在寻找:

[1, 1, 1, 2, 2, 2, 3, 3, 3]

这个更好,但需要更多时间:

numpy.concatenate([([i]*3) for i in [1,2,3]], axis=0)

答案 7 :(得分:0)

抱歉我的回答太晚了 您可以轻松使用 numpy.repeat。只需写下您想要生成的值和重复次数即可。

import numpy as np
x = [1,2,3]
y = np.linspace(0,1000,10000)
for i in x:
    new_x = np.repeat(i,len(y))
    print(new_x)