Matplotlib:在一个图中创建多个子图

时间:2017-08-03 19:31:26

标签: python-3.x pandas matplotlib

我的数据框有x1, x2, x3, x4, x5, x6, my_y列。我正在为每个xi~y做一个散点图:

%matplotlib notebook
import matplotlib.pyplot as plt
import matplotlib
matplotlib.style.use('ggplot')
my_df.plot(x='x1', y='my_y', kind = 'scatter', marker = 'x', color = 'black', ylim = [0, 10])

我为x1, x2, x3, x4, x5, x6重复了上述代码6次,创建了6个数字。我想知道是否有可能用6个散点子图制作一个数字?谢谢!

1 个答案:

答案 0 :(得分:2)

df = pd.DataFrame(
    np.random.randint(10, size=(5, 7)),
    columns='x1 x2 x3 x4 x5 x6 my_y'.split()
)

df

   x1  x2  x3  x4  x5  x6  my_y
0   0   8   3   2   7   5     8
1   0   6   2   5   8   4     9
2   4   7   1   2   6   4     5
3   8   5   4   0   5   7     4
4   5   6   0   1   8   7     2

<强>选项1
使用scatter元素中的axes方法。

fig, axes = plt.subplots(2, 3, figsize=(6, 4), sharex=True, sharey=True)
y = df.my_y.values
for i in range(6):
    axes[i//3, i%3].scatter(df.iloc[:, i].values, y)

fig.tight_layout()

enter image description here

选项2
使用pandas.DataFrame.plot

fig, axes = plt.subplots(2, 3, figsize=(6, 4), sharex=True, sharey=True)
y = df.my_y.values
for i in range(6):
    df.plot(x='x' + str(i+1),
            y='my_y',
            kind='scatter',
            marker='x',
            color='black',
            ylim=[0, 10],
            ax=axes[i//3, i%3])

fig.tight_layout()

enter image description here

对评论的回应
没有sharex=True

fig, axes = plt.subplots(2, 3, figsize=(6, 4), sharey=True)
y = df.my_y.values
for i in range(6):
    df.plot(x='x' + str(i+1),
            y='my_y',
            kind='scatter',
            marker='x',
            color='black',
            ylim=[0, 10],
            ax=axes[i//3, i%3])

fig.tight_layout()

enter image description here