主页 / 问题 / 565846
Accepted
pynista
Asked:2020-09-13 20:58:06 +0000 UTC2020-09-13 20:58:06 +0000 UTC

如何使用列表理解创建随机整数列表？

  • 772

我想创建一个包含例如二十个元素的列表。每个元素必须是 -10 到 10 之间的任意整数。

from random import randint

numbers = []
for i in range(20):
    numbers.append(randint(-10, 10))

或者像这样：

from random import randint
lst = [randint(-10, 10) for i in range(20)]

如何做同样的事情，只有在一行中（导入除外）（例如，使用列表包含、生成器表达式、函数中的一些函数、函数map()中zip()的函数itertools）并且没有中间变量i？

python

3 个回答

  1. 你可以使用 NumPy：

    import numpy as np

lst = np.random.randint(-10, 10, 20)

print(lst)

    您可以轻松创建矩阵：

    In [3]: np.random.randint(-10, 10, (5,3))
Out[3]:
array([[-10,   2,   8],
       [ -8,  -3,   8],
       [ -1,   8,   8],
       [  8,   9,  -7],
       [  4,   5,  -4]])
    • 8
  2. Best Answer
    import timeit, random, itertools, operator, functools, numpy

a, b, s = -10, 10, 1000
v1=lambda: list(numpy.random.randint(a, b, s))
v2=lambda: [random.randint(a, b) for _ in range(s)]
v3=lambda: list(map(lambda _: random.randint(a, b), range(s)))
v4=lambda: list(itertools.starmap(lambda: random.randint(a, b), [()]*s))
v5=lambda: list(r(a, b) for r in [random.randint]*s)
v6=lambda: [r() for r in [functools.partial(random.randint, a, b)]*s]
v7=lambda: list(operator.methodcaller('__call__', a, b)(r) for r in [random.randint]*s)
v8=lambda: list(map(operator.methodcaller('__call__', a, b), [random.randint]*s))
v9=lambda: list(map(operator.methodcaller('__call__'), [functools.partial(random.randint, a, b)]*s))
v10=lambda: numpy.random.randint(a, b, s).tolist()

def execTime(target_: list, repeat=1):
    for n, fn in target_: print(n, timeit.Timer(fn).timeit(1), fn())
    target_[:] = [(n, timeit.Timer(fn).timeit(repeat)) for n, fn in target_]
    for e, (n, tmt) in enumerate(sorted(target_, key=lambda r: r[1]), start=1):
        print("{}'time {} {}".format(e, n, tmt))

if __name__ == '__main__':
    target = [(n, fn) for n, fn in sorted(globals().items()) if n.startswith('v')]
    execTime(target, repeat=100)

    出去：

    v1 0.00013895335493526386 [8, 6, 8,..]
 ...
v9 0.0034144395633564986 [-8, 3, -3,..]
1'time v10 0.005898359039140072
2'time v1 0.014917592744244368
3'time v5 0.32350102439781625
4'time v2 0.33059889747340865
5'time v4 0.3355538953844154
6'time v8 0.34316236373969566
7'time v3 0.3506998448115352
8'time v6 0.3632192647513499
9'time v9 0.3795784125105248
10'time v7 0.4123779585340184
    • 7

  3. 如果您想形成一个没有重复的样本，则适用。 

    from random import sample

list = sample(range(-10, 11), 20)
    • 1

相关问题