关联规则网络图python代码
时间: 2023-07-03 20:30:15 浏览: 154
python绘制关系网络图
5星 · 资源好评率100%
下面是一个使用Python的关联规则网络图示例代码,使用了NetworkX和Matplotlib库进行绘图:
```python
import networkx as nx
import matplotlib.pyplot as plt
import pandas as pd
# 数据准备
df = pd.read_csv("关联规则.csv", encoding='gbk')
nodes = df['itemsets'].tolist()
edges = []
for i in range(len(df)):
itemset = df.iloc[i]['itemsets']
items = itemset.split(',')
for j in range(len(items)):
for k in range(j+1, len(items)):
edges.append((items[j], items[k], df.iloc[i]['lift']))
# 可视化
plt.figure(figsize=(12, 6))
G = nx.Graph()
G.add_nodes_from(nodes)
G.add_weighted_edges_from(edges)
pos = nx.spring_layout(G)
nx.draw(G, pos, with_labels=True, node_color='lightblue', node_size=1500, font_size=20, width=2, edge_color='gray')
labels = nx.get_edge_attributes(G, 'weight')
nx.draw_networkx_edge_labels(G, pos, edge_labels=labels, font_size=15)
plt.show()
```
这段代码中,我们使用了Pandas库读取了一个CSV文件,并使用NetworkX和Matplotlib库进行了关联规则的网络图可视化,绘制了一个基于Lift值的带权重边的图形。你可以根据自己的需求进行修改和扩展。
阅读全文