python - 将 dpi 添加到 seaborn 或使用给定的 dpi 导出它们

Question

我有一个可视化代码，例如:

for cluster in ready_couples_2.cluster.unique():
    sns.set(rc={'figure.figsize':(11.7,8.27)})
    # mask the cluster of interest
    is_cluster = ready_couples_2.cluster.eq(cluster)

    ax = ready_couples_2[~is_cluster].plot.scatter(x='longitude',y='latitude', c='gray') 

    ax = sns.scatterplot(data=ready_couples_2[is_cluster],
                    x='longitude', 
                    y='latitude',
                    hue='id_easy',
                    ax=ax)
    ax.legend_.remove()

    figure = ax.get_figure()    
    figure.savefig('test.png', dpi=500)

    plt.show()

但是 figure 仅保存给定图中的一个图。如何保存全部或如何在绘图时定义 DPI？

Answer 1

嗯，它确实保存了您的所有图形，但因为您没有更改文件名，所以最后一个图将是您看到的唯一图。

例如，您可以这样做

for ind, cluster in enumerate(ready_couples_2.cluster.unique()):
    ...
    figure.savefig('test%d.png' % ind, dpi=500)