python - 使用 for 循环生成多个图；在 matplotlib 子图中显示输出

Question

目标:使用 for 循环生成 100 个条形图，并将输出显示为子图图像

数据格式:具有 101 列的数据文件。最后一列是 X 变量；剩下的 100 列是 Y 变量，x 是根据它绘制的。

期望的输出:5 x 20 子图数组中的条形图，如本例图像所示:

当前方法:我一直在 seaborn 中使用 PairGrid，它生成一个 n x 1 数组: .

其中输入 == 数据帧； input3 == 从中调用列标题的列表:

for i in input3:
    plt.figure(i)
    g = sns.PairGrid(input,
             x_vars=["key_variable"],
             y_vars=i,
             aspect=.75, size=3.5)
    g.map(sns.barplot, palette="pastel")

有没有人知道如何解决这个问题？

Answer 1

举例说明如何在 20 x 5 子图的网格上绘制 100 个数据框列:

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

data = np.random.rand(3,101)
data[:,0] = np.arange(2,7,2)
df = pd.DataFrame(data)

fig, axes = plt.subplots(nrows=5, ncols=20, figsize=(21,9), sharex=True, sharey=True)
for i, ax in enumerate(axes.flatten()):
    ax.bar(df.iloc[:,0], df.iloc[:,i+1])
    ax.set_xticks(df.iloc[:,0])

plt.show()