堆叠条形图：在matplotlib中绘制堆叠条形图

一个方便使用pandas DataFrame 在matplotlib中绘制堆叠条形图的函数。

> 从 mlxtend.plotting 导入 stacked_barplot

概述

一个方便的matplotlib函数，用于从DataFrame中创建条形图，其中每个样本与多个类别相关联。

参考文献

• -

示例 1 - 来自 Pandas DataFrame 的分层条形图

import pandas as pd

s1 = [1.0, 2.0, 3.0, 4.0]
s2 = [1.4, 2.1, 2.9, 5.1]
s3 = [1.9, 2.2, 3.5, 4.1]
s4 = [1.4, 2.5, 3.5, 4.2]
data = [s1, s2, s3, s4]

df = pd.DataFrame(data, columns=['X1', 'X2', 'X3', 'X4'])
df.columns = ['X1', 'X2', 'X3', 'X4']
df.index = ['Sample1', 'Sample2', 'Sample3', 'Sample4']
df

默认情况下，DataFrame 的索引用作列标签，而 DataFrame 的列用于绘图图例。

import matplotlib.pyplot as plt
from mlxtend.plotting import stacked_barplot

fig = stacked_barplot(df, rotation=45, legend_loc='best')

API

stacked_barplot(df, bar_width='auto', colors='bgrcky', labels='index', rotation=90, legend_loc='best')

Function to plot stacked barplots

Parameters

• df : pandas.DataFrame

  A pandas DataFrame where the index denotes the x-axis labels, and the columns contain the different measurements for each row. bar_width: 'auto' or float (default: 'auto') Parameter to set the widths of the bars. if 'auto', the width is automatically determined by the number of columns in the dataset. colors: str (default: 'bgrcky') The colors of the bars. labels: 'index' or iterable (default: 'index') If 'index', the DataFrame index will be used as x-tick labels. rotation: int (default: 90) Parameter to rotate the x-axis labels.

• legend_loc : str (default: 'best')

  Location of the plot legend {best, upper left, upper right, lower left, lower right} No legend if legend_loc=False

Returns

• fig : matplotlib.pyplot figure object

Examples

For usage examples, please see https://rasbt.github.io/mlxtend/user_guide/plotting/stacked_barplot/