# -*- coding: utf-8 -*-
import numpy;
import matplotlib;
from pandas import read_csv;
from matplotlib import pyplot as plt;
font = {
'family' : 'SimHei'
};
matplotlib.rc('font', **font);
data = read_csv('F://baiduNetdisk//pytorch//5//5.4//data.csv');
d1 = '手机品牌';
d2 = '通信品牌';
v = "月消费";
gb = data.groupby(by=[d1, d2])['月消费(元)'].agg({v: numpy.sum});
d1size = gb.index.levels[0].size;
d2size = gb.index.levels[1].size;
#分组的类别数目
index = numpy.arange(d1size);
colors=['r', 'g', 'b'];
for i in range(0, d2size):
print(i);
subgb = gb[v][gb.index.labels[1]==i];
bar = plt.bar(index*d2size + i, subgb, color=colors[i]);
#将三种类型画到相同的图像中,即按照标准一个一个的化就可以
lIndex = numpy.arange(d1size)*d2size;
plt.xticks(lIndex + 3/2, gb.index.levels[0]);
plt.legend(gb.index.levels[1]);
plt.show();
画图
最新推荐文章于 2023-12-28 13:32:04 发布