如何在python代码中查找列表推导

我想在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