将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)