python绘制关联规则图
时间: 2023-10-10 08:16:07 浏览: 288
绘制关系图
Python中可以使用matplotlib和networkx库来绘制关联规则图。
首先,需要将关联规则转换为图的形式。可以将每个项作为一个节点,每个规则作为一条边。可以使用networkx库中的DiGraph类来创建一个有向图。例如,假设有以下两条规则:
{A, B} -> {C}
{B, D} -> {E}
可以将它们转换为以下图形:
![image.png](attachment:image.png)
代码示例:
```python
import networkx as nx
import matplotlib.pyplot as plt
# 创建一个有向图
G = nx.DiGraph()
# 添加节点
G.add_nodes_from(['A', 'B', 'C', 'D', 'E'])
# 添加边
G.add_edge('A', 'C')
G.add_edge('B', 'C')
G.add_edge('B', 'E')
G.add_edge('D', 'E')
# 绘制图形
nx.draw_networkx(G, with_labels=True, node_color='lightblue', node_size=1000, font_size=20, edge_color='gray')
plt.show()
```
上述代码中,使用nx.DiGraph()创建有向图,使用add_nodes_from()添加节点,使用add_edge()添加边。最后使用nx.draw_networkx()方法绘制图形,其中with_labels参数表示是否显示节点标签,node_color和node_size表示节点的颜色和大小,font_size表示节点标签的字体大小,edge_color表示边的颜色。
运行代码后,会得到如下图形:
![image-2.png](attachment:image-2.png)
可以看到,图形中每个节点代表一个项,箭头表示规则。
阅读全文