python函数可以知道它何时被列表推导调用？

内容导读

互联网集市收集整理的这篇技术教程文章主要介绍了python函数可以知道它何时被列表推导调用？，小编现在分享给大家，供广大互联网技能从业者学习和参考。文章包含849字，纯文字阅读大概需要2分钟。

内容图文

我想创建一个python函数,当它从列表推导中调用时表现不同：

def f():
    # this function returns False when called normally,
    # and True when called from a list comprehension
    pass

>>> f()
False
>>> [f() for _ in range(3)]
[True, True, True]

我试着查看inspect模块,dis模块和lib2to3的解析器,以便使这个技巧有效,但是没有找到任何东西.还有一个简单的原因,为什么这个不存在,我没有想到.

解决方法:

您可以通过以下方式检查堆栈帧来确定这一点：

def f():
    try:
        raise ValueError
    except Exception as e:
        if e.__traceback__.tb_frame.f_back.f_code.co_name == '<listcomp>':
            return True

然后：

>>> print(f())
None
>>> print([f() for x in range(10)])
[True, True, True, True, True, True, True, True, True, True]

但不建议这样做.真的,不是.

注意

目前,这仅按要求检测列表推导.它不会检测发电机的使用.例如：

>>> print(list(f() for x in range(10)))
[None, None, None, None, None, None, None, None, None, None]

内容总结

以上是互联网集市为您收集整理的python函数可以知道它何时被列表推导调用？全部内容，希望文章能够帮你解决python函数可以知道它何时被列表推导调用？所遇到的程序开发问题。 如果觉得互联网集市技术教程内容还不错，欢迎将互联网集市网站推荐给程序员好友。

内容备注

版权声明：本文内容由互联网用户自发贡献，该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 gblab@vip.qq.com 举报，一经查实，本站将立刻删除。

内容手机端