matplotlib中的集群条形图?

时间:2014-02-16 12:04:14

标签: python matplotlib

如何绘制类似于

的条形图

Clustered bar plot in gnuplot使用python matplotlib?

date|name|empid|app|subapp|hours
20140101|A|0001|IIC|I1|2.5
20140101|A|0001|IIC|I2|3
20140101|A|0001|IIC|I3|4
20140101|A|0001|CAR|C1|2.5
20140101|A|0001|CAR|C2|3
20140101|A|0001|CAR|C3|2
20140101|A|0001|CAR|C4|2

尝试使用app为同一个人绘制子应用小时数。无法在matplotlib的演示页面中看到一个示例。

编辑:下面引用的例子似乎都不适用于上述每个类别的不等数量的条形码。

1 个答案:

答案 0 :(得分:1)

示例没有管理不等数量的条形图,但您可以使用其他方法。我会给你一个例子。

注意:我使用pandas 操纵你的数据,如果你不知道它,你应该试一试http://pandas.pydata.org/

import pandas as pd
import matplotlib.pyplot as plt
import matplotlib.ticker as mtick
import numpy as np

df = pd.read_table("data.csv",sep="|")
grouped = df.groupby('app')['hours']

colors = "rgbcmyk"

fig, ax = plt.subplots()
initial_gap = 0.1
start = initial_gap
width = 1.0
gap = 0.05
for app,group in grouped:
    size = group.shape[0]
    ind = np.linspace(start,start + width, size+1)[:-1]   
    w = (ind[1]-ind[0])
    start = start + width + gap
    plt.bar(ind,group,w,color=list(colors[:size]))

tick_loc = (np.arange(len(grouped)) * (width+gap)) + initial_gap + width/2
ax.set_xticklabels([app for app,_ in grouped])
ax.xaxis.set_major_locator(mtick.FixedLocator(tick_loc))

plt.show()

data.csv 数据

date|name|empid|app|subapp|hours
20140101|A|0001|IIC|I1|2.5
20140101|A|0001|IIC|I2|3
20140101|A|0001|IIC|I3|4
20140101|A|0001|CAR|C1|2.5
20140101|A|0001|CAR|C2|3
20140101|A|0001|CAR|C3|2
20140101|A|0001|CAR|C4|2