如何简化Python中分布相关的列表推导式（避免重复调用getattr）

嘿，这个问题很好解决！你原来的代码里两次调用getattr确实存在冗余——每次循环都要重复获取同一个属性值，我们可以通过先一次性预获取所有属性值的方式来优化，下面是两种简洁的解决方案：

方案1：使用生成器表达式预获取属性（推荐） #

这种方式可读性强，同时保证每个属性只调用一次getattr：

import scipy.stats as st

# 先通过生成器一次性获取所有属性值，再过滤连续分布对象
CONTINUOUS_DISTRIBUTIONS = [dist for dist in (getattr(st, d) for d in dir(st)) if isinstance(dist, st.rv_continuous)]

print('number of distributions = ', len(CONTINUOUS_DISTRIBUTIONS))
print('CONTINUOUS_DISTRIBUTIONS = ')
print(CONTINUOUS_DISTRIBUTIONS)

原理说明 #

这里的(getattr(st, d) for d in dir(st))是一个生成器表达式，它会遍历scipy.stats的所有属性名，只调用一次getattr获取每个属性的值，之后列表推导式只需要检查这个预获取的值是否是st.rv_continuous的实例即可，完美避免了重复调用。

方案2：结合filter函数实现 #

如果你更喜欢函数式风格，也可以用filter配合生成器表达式来实现：

import scipy.stats as st

CONTINUOUS_DISTRIBUTIONS = list(filter(lambda dist: isinstance(dist, st.rv_continuous), (getattr(st, d) for d in dir(st))))

# 后续打印代码和之前一致
print('number of distributions = ', len(CONTINUOUS_DISTRIBUTIONS))
print('CONTINUOUS_DISTRIBUTIONS = ')
print(CONTINUOUS_DISTRIBUTIONS)

两种方案都能达到只调用一次getattr的目的，其中方案1的列表推导式可读性更强，更符合Python的惯用写法。

内容的提问来源于stack exchange，提问作者Marco Aurélio Falcão