在理解另一个类变量时引用类变量

Question

这可能是一个简单的问题,但我无法进行独特的搜索.

我有一个定义静态字典的类,然后尝试静态地定义该字典的子集.

所以,作为玩具的例子:

class example(object): 
    first_d = {1:1,2:2,3:3,4:4} 
    second_d = dict((k,first_d[k]) for k in (2,3))

这产生了 NameError: global name 'first_d' is not defined

我该怎么做这个参考？似乎这种模式适用于其他情况,例如:

class example2(object):
    first = 1
    second = first + 1

Answer 1

基本列表推导具有以下语法

[expression for var in iterable]

当在类中发生列表推导时,可以使用该类的属性iterable.在Python2和Python3中也是如此.

但是,可以expression在Python2中使用(即访问)类的属性,但不能在Python3中使用.

生成器表达式的故事有点不同:

(expression for var in iterable)

虽然仍可以从中访问iterable类属性,但无法访问类属性expression.(这适用于Python2和Python3).

这可以总结如下:

                             Python2      Python3
Can access class attributes
--------------------------------------------------
list comp. iterable                Y            Y
list comp. expression              Y            N
gen expr. iterable                 Y            Y
gen expr. expression               N            N
dict comp. iterable                Y            Y
dict comp. expression              N            N

(在这方面,Dict理解与生成器表达式的行为相同.)

现在,这与您的问题有什么关系:

在你的例子中,

second_d = dict((k,first_d[k]) for k in (2,3))

一NameError发生的原因first_d是没有从访问expression的发电机表达式的一部分.

Python2的解决方法是将生成器表达式更改为列表解析:

second_d = dict([(k,first_d[k]) for k in (2,3)])

但是,我发现这不是一个非常舒服的解决方案,因为此代码在Python3中会失败.

你可以像Joel Cornett建议的那样做:

second_d = {k: v for k, v in first_d.items() if k in (2, 3)}

因为这使用first_d的iterable,而不是expression在字典理解的一部分.但是,如果first_d包含许多项目,这可能会循环通过不必要的项目.Neverthess,这个解决方案可能会很好,如果first_d很小的话.

通常,您可以通过定义可在类内部或外部定义的辅助函数来避免此问题:

def partial_dict(dct, keys):
    return {k:dct[k] for k in keys}

class Example(object):
    first_d = {1:1,2:2,3:3,4:4}
    second_d = partial_dict(first_d, (2,3))

class Example2(object):
    a = [1,2,3,4,5]
    b = [2,4]
    def myfunc(A, B):
        return [x for x in A if x not in B]
    c = myfunc(a, b)

print(Example().second_d)
# {2: 2, 3: 3}

print(Example2().c)
# [1, 3, 5]

函数起作用是因为它们定义了局部范围,并且可以从字典理解中访问此局部范围中的变量.

这里解释了这一点,但我对此并不完全满意,因为它没有解释为什么该expression部分的行为与iterable列表理解,生成器表达或字典理解的部分不同.

因此,我无法(完全)解释为什么 Python会以这种方式运行,只是这就是它的行为方式.