python获取ast并存储为邻接矩阵

要获取Python代码的 AST（抽象语法树）并将其转换为邻接矩阵，你可以使用 Python 的标准库 `ast` 和 `numpy` 库。首先，你需要将代码转换为 AST，然后遍历 AST 并提取节点之间的关系，最后将其表示为邻接矩阵。下面是一个示例代码：

import ast
import numpy as np

# 读取 Python 代码
with open('example.py', 'r') as f:
    code = f.read()

# 获取 AST
tree = ast.parse(code)

# 获取所有节点
nodes = [node for node in ast.walk(tree)]

# 获取节点之间的关系
adj_matrix = np.zeros((len(nodes), len(nodes)))
for i, node1 in enumerate(nodes):
    for j, node2 in enumerate(nodes):
        if i != j and issubclass(type(node1), ast.AST) and issubclass(type(node2), ast.AST):
            if any((node1 is child or node2 is child) for child in ast.iter_child_nodes(node1)):
                adj_matrix[i][j] = 1
                adj_matrix[j][i] = 1

# 输出邻接矩阵
print(adj_matrix)

在上面的示例代码中，我们首先读取 Python 代码，然后使用 `ast.parse()` 函数将其转换为 AST。然后我们遍历 AST 并提取节点之间的关系。我们首先创建一个大小为（节点数，节点数）的零矩阵，并使用 `enumerate()` 函数遍历所有节点。对于每对节点，我们检查它们是否有父子关系，如果是，则在邻接矩阵中将它们的对应项设置为 1。最后，我们输出邻接矩阵。

请注意，上面的示例代码只是一个示例，可能需要根据实际需求进行修改。例如，我们可以根据需要添加更多的节点类型，或者根据需要使用不同的邻接矩阵表示方法。