graph = nx.DiGraph() graph.add_nodes_from(example.dominoModules[i].nodes()) graph.add_edges_from(example.dominoModules[i].edges()) pos = nx.spring_layout(graph) plt.title(f'Bayesian Network Based On {example.dominoModules[i].get_roots()[0]}') nx.draw(graph, pos, with_labels=True, node_color='lightblue', node_size=2000, edge_color='gray', arrows=True) plt.show()改成使用multipartite_layout绘制

要使用`nx.multipartite_layout()`绘制贝叶斯网络图，您需要为每个节点指定层次。由于您没有提供具体的节点层次信息，我将提供一个示例代码，其中假设根节点在最上面，一级节点在根节点的下面，以此类推。请注意，您需要根据实际情况调整节点的层次和样式。

import matplotlib.pyplot as plt
import networkx as nx
from pgmpy.readwrite import XMLBIFReader

# 从XML文件中读取贝叶斯网络
reader = XMLBIFReader("network.xml")
model = reader.get_model()

# 创建一个空的有向图
G = nx.DiGraph()

# 添加节点和层次信息
nodes = model.nodes()
levels = {}
for node in nodes:
    level = model.get_independencies(node).level()
    levels[node] = level
    G.add_node(node, level=level)

# 添加边
G.add_edges_from(model.edges())

# 设置布局
pos = nx.multipartite_layout(G, subset_key="level", align="vertical")

# 绘制图形
nx.draw(G, pos, with_labels=True, node_size=1000, node_color="lightblue", font_size=10)

# 显示图形
plt.axis("off")
plt.show()

在这个示例中，我们首先从XML文件中读取贝叶斯网络模型，并创建一个空的有向图。然后，我们为每个节点添加层次信息，并将其作为节点属性存储在图中。接下来，我们添加边，然后使用`nx.multipartite_layout()`函数根据节点的层次信息设置节点的布局，使得根节点在最上面，一级节点在根节点的下面，以此类推。最后，使用`nx.draw()`函数绘制图形，并使用`plt.axis("off")`和`plt.show()`函数显示图形。

请注意，您需要将上述代码中的`"network.xml"`替换为您自己的贝叶斯网络模型的文件名或路径。此外，您可能需要根据实际情况调整节点大小、颜色、字体大小等参数以满足您的需求。