python - Seaborn:如何在 catplot 中设置自定义 "hue"标签而不是自动标签

Question

我正在使用 seaborn 绘制漂亮的分类图。 seaborn.catplot 对我帮助很大。它使用 pandas.DataFrame 的列名作为标签，但有时我想在我的绘图上重命名这些标签。

很明显，我可以使用 matplotlib.pyplot 来设置轴标签、标题等。但是我怎样才能改变色调标签呢？我不希望 ugly_column_name 显示在我的绘图上。我想要一个不同的字符串，例如 Pretty Category。

这是我的情节的极其简化的代码，以及生成的图像。我如何在这里编辑色调标签？有没有简单的一行？

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

data = pd.DataFrame({"ugly_column_name": ["foo", "foo", "foo", "bar", "bar", "bar"],
                     "x": [1, 2, 3, 1, 2, 3],
                     "y": [1, 1.1, 1.5, 1.1, 1.5, 2.0]}
                    )
sns.catplot(x="x", y="y", data=data, hue="ugly_column_name")
plt.xlabel('More than just lowerscore x')

Answer 1

您可以使用 Dataframe.replace

在您的情况下，将此行添加到您的代码中:

data = data.replace(to_replace="foo", value="foo_alternative")

请检查代码段:

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

data = pd.DataFrame({"ugly_column_name": ["foo", "foo", "foo", "bar", "bar", "bar"],
                     "x": [1, 2, 3, 1, 2, 3],
                     "y": [1, 1.1, 1.5, 1.1, 1.5, 2.0]}
                    )

data = data.replace(to_replace="foo", value="foo_alternative")
data = data.replace(to_replace="bar", value="bar_alternative")

sns.catplot(x="x", y="y", data=data, hue="ugly_column_name")
plt.xlabel('More than just lowerscore x')