如何在多个子地块中绘制seaborn lmplots

2024-09-27 07:31:39 发布

您现在位置:Python中文网/ 问答频道 /正文

我试图在同一个图中绘制多个lmplots。但是我得到了太多不需要的子图

我找到了另一个SO链接How to plot 2 seaborn lmplots side-by-side?,但这对我也没有帮助

在本例中,我需要1行2列

MWE

# imports
import numpy as np
import pandas as pd
import seaborn as sns
import matplotlib.pyplot as plt

# data
df = sns.load_dataset('titanic')

# plot
m,n = 1,2
figsize=(12,8)
cols1 = ['age','fare']
cols2 = ['fare','age']
target = 'survived'
fontsize = 12

fig, ax = plt.subplots(m,n,figsize=figsize)
for i, (col1,col2) in enumerate(zip(cols1,cols2)):
    plt.subplot(m,n,i+1)

    sns.lmplot(x=col1,y=col2,data=df,
           hue=target, palette='Set1',
           scatter_kws={'alpha':0.3})

    plt.xlabel(col1,fontsize=fontsize)
    plt.ylabel(col2,fontsize=fontsize)

    plt.tick_params(axis='both', which='major', labelsize=fontsize)
    plt.tight_layout()

for i in range(m*n-len(cols1)):
    ax.flat[-(i+1)].set_visible(False)

Tags: importdfdataplotaspltseabornside
1条回答
网友
1楼 · 发布于 2024-09-27 07:31:39

我迄今为止的努力:

df = pd.DataFrame({'x0':[10,20,30,40],
                   'y0': [100,200,300,400],
                   'x1':[0.1,0.2,0.3,0.1],
                   'y1':[0.01,0.02,0.03,0.01],
                   'target': [0,1,1,1]
                   })

df1 = df.append(df)
df1 = df1.reset_index(drop=True)
df1['x0'].iloc[len(df):] = df['x1'].to_numpy()
df1['y0'].iloc[len(df):] = df['y1'].to_numpy()
df1['col'] = ['c0']* len(df) + ['c1'] * len(df)
df1 = df1.drop(['x1','y1'],axis=1)
df1 = df1.rename(columns={'x0':'x','y0':'y'})

sns.lmplot(x='x',y='y',hue='target',data=df1,col='col')

输出: enter image description here

相关问题 更多 >

    热门问题