标签：python list-comprehension pygments

我想在python源代码中找到一个列表理解,为此我尝试使用Pygments,但它没有找到这样做的方法.

更具体地说,我想做一个能识别所有可能的列表理解的函数.例如：

[x**2 for x in range(5)]

[x for x in vec if x >= 0]

[num for elem in vec for num in elem]

[str(round(pi, i)) for i in range(1, 6)]

这个例子是从https://docs.python.org/2/tutorial/datastructures.html#list-comprehensions获得的

它也是一个有正则表达式的解决方案.

谢谢

解决方法:

您可以使用ast库将Python代码解析为语法树,然后遍历解析后的树以查找ListComp表达式.

这是一个简单的例子,用于打印在通过stdin传递的Python代码中找到列表推导的行号：

import ast
import sys

prog = ast.parse(sys.stdin.read())
listComps = (node for node in ast.walk(prog) if type(node) is ast.ListComp)
for comp in listComps:
    print "List comprehension at line %d" % comp.lineno

标签：python,list-comprehension,pygments
来源： https://codeday.me/bug/20190611/1220539.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。