Matplotlib 中的叠加条形图
我们在 Matplotlib 中使用 matplotlib.pyplot.bar()
方法生成条形图。为了将某个数据集的条形图堆叠在另一个数据集上,我们将所有需要堆叠的数据集相加,并将总和作为 bottom
参数传递给 bar()
方法。
import matplotlib.pyplot as plt
data1=[30,20,10,0,0]
data2=[20,20,20,20,0]
data3=[50,60,70,80,100]
year=["2015","2016","2017","2018","2019"]
fig,ax=plt.subplots(3,1,figsize=(10,8))
ax[0].bar(year,data1,color="red")
ax[0].legend(["C++"])
ax[1].bar(year,data2,color="yellow")
ax[1].legend(["JavaScript"])
ax[2].bar(year,data3,color="green")
ax[2].legend(["Python"])
plt.show()
输出:
在这里,我们有三个独立的条形图,分别代表了某公司员工五年来对一种编程语言的偏好。我们将讨论如何将一门语言的条形图叠加到另一门语言上,并通过一张条形图来研究多年来人们对编程语言的整体选择。
叠加条形图 Matplotlib
import numpy as np
import matplotlib.pyplot as plt
data1=[30,20,10,0,0]
data2=[20,20,20,20,0]
data3=[50,60,70,80,100]
year=["2015","2016","2017","2018","2019"]
plt.figure(figsize=(9,7))
plt.bar(year,data3,color="green",label="Python")
plt.bar(year,data2,color="yellow",bottom=np.array(data3),label="JavaScript")
plt.bar(year,data1,color="red",bottom=np.array(data3)+np.array(data2),label="C++")
plt.legend(loc="lower left",bbox_to_anchor=(0.8,1.0))
plt.show()
输出:
它将一个条形图叠加在另一个条形图上。在图中,我们先将 data3
绘制成 Python 数据,作为其他条形图的基础,然后再绘制 data2
的条形图,将 data3
的条形图作为 data2
条形图的基础。为了将 data2
的条形图叠加到 data3
上,我们设置 bottom=np.array(data3)
。
同样,在绘制 data1
的条形图时,我们以 data2
和 data3
的条形图为基础。为此,我们在绘制 data1
的条形图时,设置 bottom=np.array(data3)+np.array(data2)
。
需要注意的是,我们必须使用 NumPy
数组来添加 bottom
参数的数据。如果我们设置 bottom=data3+data2
,它将通过在列表 data3
的末尾追加 data2
的元素来创建列表。
如果我们不想使用 NumPy
数组,我们可以使用列表推导法来添加 list 的对应元素。
import numpy as np
import matplotlib.pyplot as plt
data1=[30,20,10,0,0]
data2=[20,20,20,20,0]
data3=[50,60,70,80,100]
year=["2015","2016","2017","2018","2019"]
plt.figure(figsize=(9,7))
plt.bar(year,data3,color="green",label="Python")
plt.bar(year,data2,color="yellow",bottom=data3,label="JavaScript")
plt.bar(year,data1,color="red",bottom=[sum(data) for data in zip(data2,data3)],label="C++")
plt.legend(loc="lower left",bbox_to_anchor=(0.8,1.0))
plt.show()
输出:
使用 Pandasd 的 Matplotlib 叠加条形图的方法
我们也可以使用 Python 中的 Pandas
库,在 Python 中生成堆叠的柱状图。
import pandas as pd
import matplotlib.pyplot as plt
years=["2015","2016","2017","2018","2019"]
data={
"Python":[50,60,70,80,100],
"JavaScript":[20,20,20,20,0],
"C++":[30,20,10,0,0],
}
df=pd.DataFrame(data,index=years)
df.plot(kind="bar",stacked=True,figsize=(10,8))
plt.legend(loc="lower left",bbox_to_anchor=(0.8,1.0))
plt.show()
输出:
它从 Pandas DataFrame 中生成一个堆叠的条形图,其中一列的条形图堆叠在 DataFrame 中每个索引的另一列上。
Suraj Joshi is a backend software engineer at Matrice.ai.
LinkedIn