import numpy as np import matplotlib.pyplot as plt import pandas as pd dataset = pd.read_csv('E:\Program Files (x86)\Apriori-Association-Rules-main\Market_Basket_Optimisation.csv', header = None) print(dataset)运行后,用条形图显示各类商品的累计购买量应该如何写代码
时间: 2024-03-13 20:48:20 浏览: 68
你可以按照以下步骤绘制条形图:
1. 统计每种商品被购买的次数。
2. 对商品按照购买次数进行排序。
3. 取出前十个购买次数最多的商品。
4. 使用 Matplotlib 库的 `bar()` 函数绘制条形图。
下面是实现代码:
```python
import numpy as np
import matplotlib.pyplot as plt
import pandas as pd
# 读入数据集
dataset = pd.read_csv('E:\Program Files (x86)\Apriori-Association-Rules-main\Market_Basket_Optimisation.csv', header=None)
# 统计每种商品被购买的次数
items = {}
for row in dataset.values:
for item in row:
if item not in items:
items[item] = 1
else:
items[item] += 1
# 对商品按照购买次数进行排序
sorted_items = sorted(items.items(), key=lambda x: x[1], reverse=True)
# 取出前十个购买次数最多的商品
top_items = sorted_items[:10]
# 绘制条形图
plt.bar(range(len(top_items)), [item[1] for item in top_items], tick_label=[item[0] for item in top_items])
plt.show()
```
该代码会绘制出前十个购买次数最多的商品的条形图,横坐标是商品名称,纵坐标是购买次数。你可以根据需要修改代码中数据集的文件路径、显示的商品数量等参数。
阅读全文