如何使用 matplotlib 到 plot 复杂条形图 - 具有多行条形的多个子图？

Question

我有一个示例 dataframe，如下所示：

import pandas as pd

df = pd.DataFrame(np.random.randint(
    0, 10, size=(1000, 11)), columns=list('ABCDEFGHIJK'))

所需但未抛光的 output 如下所示：

dataframe 中每一列的数据被绘制为具有五行条形的子图。

我更喜欢使用 matplotlib 因为我可以相对轻松地使图表看起来不错。 但它的性能似乎相当缓慢。

Answer 1

您可以使用bar的bottom参数来偏移各个行。
以下未优化的示例演示了这种方法：

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

df = pd.DataFrame(np.random.randint(0, 10, size=(1000, 11)), columns=list('ABCDEFGHIJK'))

fig = plt.figure()

for i,c in enumerate(df.columns):
    ax = fig.add_subplot(3, 4, i+1)
    for x,h,b in zip((df.index.to_numpy() % 200).reshape(-1, 200), df[c].to_numpy().reshape(-1, 200), (df.index.to_numpy() // 200 * 10).reshape(-1, 200)):
        ax.set_title(c)
        ax.bar(x, h, bottom=b, color='k' )