gpt4 book ai didi

python - 为什么 Seaborn 绘制了两个图例,如何删除一个并修复另一个?

转载 作者:行者123 更新时间:2023-12-01 00:41:18 25 4
gpt4 key购买 nike

当我运行下面所示的代码时,我得到一个包含 2 个图例的图形。我不明白为什么要绘制两个,而且我无法删除其中一个。我的目标是保留图形外部的图例,删除图形内部的图例,并以某种方式阻止奇怪的裁剪,即切断图形外部图例的右侧。 figure showing the two legends with random data

我之前有一个question询问类似的问题,但该问题是通过使用 seaborns 散点图而不是 relplot 解决的。遗憾的是,在该问题中有效的答案在这里都不起作用。如果这个问题是由于我试图绘制的图形类型的“非常规”方式引起的,那么请告诉我。正确地做比通过破解来找到解决方案更好......

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


#setup
sns.set(font_scale=2)
sns.set_context('poster')

#figure and axes
fig = plt.figure(figsize=(20,20))
axs = {i:fig.add_subplot(330+i) for i in range(1,10)}




#create random data
r = np.random.randint
N=10
df = pd.DataFrame(columns=['No.','x1','x2','x3','y1','y2','y3'])
for i in range(N):
df.loc[i] = i+1,r(50,high=100),r(50,high=100),r(50,high=100),r(50,high=100),r(50,high=100),r(50,high=100)

#create axes labels
x_labels = ['x1','x2','x3']
y_labels = ['y1','y2','y3']
xy_labels = [(x,y) for y in y_labels for x in x_labels ]


#plot on axes
for i,(x_label,y_label) in enumerate(xy_labels):
if i ==0:#if statement so only one of the plots has legend='full'
a = sns.scatterplot(
data=df,
x=x_label,
y=y_label,
legend='full', #create the legend
ax=axs[i+1],
hue='No.',
palette=sns.color_palette("hls", N)
)

fig.legend(bbox_to_anchor=(1, 0.7), loc=2, borderaxespad=0.) #Move the legend outside the plot
a.legend_.remove() #attempt to remove the legend
else:
a = sns.scatterplot(
data=df,
x=x_label,
y=y_label,
legend=False,
ax=axs[i+1],
hue='No.',
palette=sns.color_palette("hls", N)
)

#remove axes labels from specific plots
if i not in [0,3,6]: axs[i+1].set_ylabel('')
if i not in [6,7,8]: axs[i+1].set_xlabel('')


#add line plots and set limits
for ax in axs.values():
sns.lineplot(x=range(50,100),y=range(50,100), ax=ax, linestyle='-')
ax.set_xlim([50,100])
ax.set_ylim([50,100])


fig.tight_layout()

最佳答案

您可以在代码的最后部分添加 legend=False

#setup
sns.set(font_scale=2)
sns.set_context('poster')

#figure and axes
fig = plt.figure(figsize=(20,20))
axs = {i:fig.add_subplot(330+i) for i in range(1,10)}

#create axes labels
x_labels = ['x1','x2','x3']
y_labels = ['y1','y2','y3']
xy_labels = [(x,y) for y in y_labels for x in x_labels ]

#plot on axes
for i,(x_label,y_label) in enumerate(xy_labels):
if i ==0:#if statement so only one of the plots has legend='full'
a = sns.scatterplot(
data=df,
x=x_label,
y=y_label,
legend='full', #create the legend
ax=axs[i+1],
hue='No.',
palette=sns.color_palette("hls", N)
)
fig.legend(bbox_to_anchor=(1, 0.7), loc=2, borderaxespad=0.) #Move the legend outside the plot
a.legend_.remove() #attempt to remove the legend
else:
a = sns.scatterplot(
data=df,
x=x_label,
y=y_label,
legend=False,
ax=axs[i+1],
hue='No.',
palette=sns.color_palette("hls", N)
)

#remove axes labels from specific plots
if i not in [0,3,6]: axs[i+1].set_ylabel('')
if i not in [6,7,8]: axs[i+1].set_xlabel('')

#add line plots and set limits
for ax in axs.values():
sns.lineplot(x=range(50,100),y=range(50,100), ax=ax, linestyle='-', legend=False)
ax.set_xlim([50,100])
ax.set_ylim([50,100])

fig.tight_layout()

结果:

enter image description here

关于python - 为什么 Seaborn 绘制了两个图例,如何删除一个并修复另一个?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/57326558/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com