Python:list comprehension,它引用了一个内部生成的列表

时间:2013-10-15 16:45:08

标签: python

首先,如果您将此问题分为三个步骤,则无疑问题:1)创建some_list 2)创建random_list和3)对两个列表(即[(x, y, f(y, m)) for x in l for y in m])运行直接列表理解。由于我正在使用的一些API的限制,我希望能够用两行编写代码:1)创建some_list 2)运行列表理解,创建第二个第二个列表并允许一些操作/函数/方法内部生成的列表本身。

这是我希望能够做到的:

import random
[(x, y, not random_list.index(y)) for x in some_list
 for y in random.sample([1,2,3,4,5], (random.choice([1,2,3]))) as random_list]

我知道as random_list在这里不起作用。我知道如果我在编写列表解析之前基于random.sample([1,2,3,4,5], random.choice([1,2,3]))创建一个列表,这里没有任何问题。

写出问题会得出一些答案。以下是上述具体问题的具体答案:

[(x, y, not i) for x in some_list
 for i, y in enumerate(random.sample([1,2,3,4,5], random.choice([1,2,3])))]

我的基本问题是:如果列表是在列表推导中创建的,那么该列表本身是否可以在列表推导中引用?或者我需要创建某种包装器吗?

def wrapper():
    f = lambda i, l: 42
    l = random_list()
    return [(i, f(i, l)) for i in l]

[(x, i, v) for x in some_list for i, v in wrapper()]

我想我已基本回答了我的问题,但这似乎是一个值得写的好文章。如果有其他想法,评论,那将是非常有用的。

@DSM有效问题。这是特定的(django)代码:

PlayerPosition.objects.bulk_create([
    PlayerPosition(
        player=player,
        position_id=position_id,
        primary=not index
    ) for player in Player.objects.all()
     for index, position_id in enumerate(random_position_list())
])

2 个答案:

答案 0 :(得分:4)

基本上,您希望创建一个列表,将其绑定到一个名称,然后在列表解析中使用该名称。阻止在表达式中绑定名称的原因是赋值是语句。但是,您可以(ab)使用lambda来获得您想要的内容:

(lambda random_list: [
    (x, y, not random_list.index(y)) 
    for x in some_list
    for y in random_list])(
        random_list=random.sample([1, 2, 3, 4, 5], random.choice([1, 2, 3])))

这是因为它是一行代码,但我不确定它是否适用于你的其他约束,这些约束在这一点上是神秘而不清楚的。


编辑:如果你可以做一个辅助功能,那么你几乎肯定会采取这种方法。不过,我会以不同于你的方式做到这一点。我会像这样做一个通用的:

def iter_with_list(l):
    for element in l:
        yield (l, element)

像这样使用它:

[(x, y, not random_list.index(y))
 for x in some_list
 for random_list, y in iter_with_list(random.sample([1,2,3,4,5], random.choice([1, 2, 3])))]

通过这种方式,您仍然可以将所有逻辑保留在理解中,而不会将其传播到所有地方。这只是将名称绑定为表达式的一部分的另一个技巧 - 在这种情况下,通过重复提供它来理解每次迭代的绑定 - 但它比前一个更优雅。

答案 1 :(得分:1)

  

我的基本问题是:如果列表是在列表推导中创建的,那么该列表本身是否可以在列表推导中引用?

简短的回答是,是的,它可以。您可以根据需要在列表推导中使用尽可能多的列表推导,但事情将变得非常难以理解。示范:

>>> [var for var in [val for val in range(1, 100)]]
[1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99]
>>> 

如果要组合两个列表,如元组:

您可以这样做:

>>> from random import randint
>>> zip([x for x in xrange(0, 10)], [randint(0, 100) for _ in xrange(10)])
[(0, 56), (1, 60), (2, 7), (3, 29), (4, 85), (5, 76), (6, 95), (7, 91), (8, 40), (9, 4)]

您甚至可以对列表推导的值应用操作(但它真的不可读):

>>> weird_list = [val for val in [var for var in xrange(0, 100) if not var % 2] if not val % 3]
>>> print weird_list
[0, 6, 12, 18, 24, 30, 36, 42, 48, 54, 60, 66, 72, 78, 84, 90, 96]

关于你的评论:

>>> another_weird_list = [(var, 42) for var in xrange(5)]
>>> another_weird_list
[(0, 42), (1, 42), (2, 42), (3, 42), (4, 42)]

不需要lambda。

回应你的第二条评论:

最初:

arbitrary_list = [x for x in range(10)]
f = lambda x: [var + 1 for var in x]
no_list = [(var, f(arbitrary_list)) for var in arbitrary_list]

然后:

no_list = [(var, [var + 1 for var in [x for x in range(10)]]) for var in [x for x in range(10)]]

一个班轮。如果您愿意,您仍然可以制作arbitrary_list