Skip to main content
 首页 » 编程设计

python之计算文件中 if 语句的数量

2025年05月04日20webabcd

我正在尝试读入一个代码文件并计算 if 语句的数量。我怎样才能使用 AST 来做这件事。

请您参考如下方法:

Python 有一个标准的库ast:

# assume you are reading Python source 
import ast 
 
with open('src.py') as f: 
    src = f.read() 
 
tree = ast.parse(src) 
cnt_if = sum([x for x in ast.walk(tree) if isinstance(x, ast.If)])