使用编译器AST()函数在Python中生成代码统计信息

在Python中，可以使用编译器模块（compiler）提供的ast函数来生成抽象语法树（AST）。AST是源代码的结构化表示，它以树状的形式展示了代码的各个节点和它们之间的关系。根据抽象语法树，我们可以分析、修改甚至生成代码。

首先，我们需要导入编译器模块以及相关的类和函数：

import compiler
from compiler.ast import *

然后，我们可以使用compiler.parse函数将源代码解析成AST：

source_code = """
x = 1
y = 2
z = x + y
print(z)
"""

tree = compiler.parse(source_code)

上述代码中，compiler.parse函数将源代码中的字符串解析为AST，并将其返回为根节点赋值给tree变量。

AST的树状结构由一系列节点构成，每个节点的具体类型决定了其在语法树中所代表的内容。例如，Assign节点表示赋值语句，Add节点表示加法运算，Printnl节点表示打印语句等。

我们可以使用compiler.ast.walk函数遍历AST的各个节点，统计特定类型的节点的出现次数，从而获取代码的统计信息。以下是一个统计赋值语句的例子：

assign_count = 0

def count_assigns(node):
    global assign_count
    if isinstance(node, Assign):
        assign_count += 1
    for child in node.getChildNodes():
        count_assigns(child)

count_assigns(tree)
print("Total assigns:", assign_count)

在上述例子中，我们使用一个全局变量assign_count来记录赋值语句的数量。通过定义一个递归函数count_assigns来遍历AST节点，并在遇到Assign节点时将assign_count加一。最后，打印出赋值语句的总数。

除了统计赋值语句的例子，我们还可以使用类似的方式统计其他类型的节点，如打印语句、函数定义、循环语句等。

总结起来，使用compiler.ast模块中的函数和类，结合递归遍历AST的方式，我们可以生成代码的统计信息，如语句数量、函数数量等。这些统计信息可以帮助我们更好地理解和分析代码，以及进行代码重构和优化。