如何更改 regplot() 的点大小，seaborn 的散点图函数 (python)

Question

我希望能够在这样绘图时设置点大小：

sns.regplot(y=[1,3,4,2,5], x=[range(5)], data=df,
            marker='o', color='red')
plt.show()

大家知道怎么做吗？

Answer 1

为此，您可以像这样为regplot()函数提供scatter_kws参数：

import seaborn as sns
tips = sns.load_dataset('tips')
sns.regplot(x='total_bill', y='tip', data=tips,
            marker='o', color='red', scatter_kws={'s':2})

sns.regplot(x='total_bill', y='tip', data=tips,
            marker='o', color='red', scatter_kws={'s':20})

Answer 2

您甚至可以动态调整点的大小以表示第三维。 此代码使用与 OP 相同的数据，但将其包装在 DataFrame 中（因为 seaborn 就是为此而设计的）并添加了第三维 z。

import seaborn as sns
import pandas as pd

data = pd.DataFrame({
    'x': [x for x in range(5)],
    'y': [1, 3, 4, 2, 5],
    'z': [14, 14, 100, 16, 36]
})
sns.regplot(x='x', y='y', data=data, marker='o', color='red',
    scatter_kws={'s': data['z']})

您可能可以想象如何以编程方式操作大小列表/数组，从而为您提供大量传达额外信息的能力。

Answer 3

我想补充 mburke05 的答案，似乎可以将类似数组的数据传递到scatter_kws. 例如，如果您希望tips 数据集中的size属性来确定一个点的大小，您可以这样写：

sns.regplot(
    x="total_bill", y="tip", data=tips,
    marker='o', color='red', scatter_kws={'s':tips['size']})

但是，您必须在数据框中显式查找该属性（如上所述）； 您不能像设置x和y那样简单地使用列名。