วิธีบันทึกพล็อตของ Seaborn ลงในไฟล์

ฉันลองรหัสต่อไปนี้ ( test_seaborn.py):

import matplotlib
matplotlib.use('Agg')
import matplotlib.pyplot as plt
matplotlib.style.use('ggplot')
import seaborn as sns
sns.set()
df = sns.load_dataset('iris')
sns_plot = sns.pairplot(df, hue='species', size=2.5)
fig = sns_plot.get_figure()
fig.savefig("output.png")
#sns.plt.show()

แต่ฉันได้รับข้อผิดพลาดนี้:

  Traceback (most recent call last):
  File "test_searborn.py", line 11, in <module>
    fig = sns_plot.get_figure()
AttributeError: 'PairGrid' object has no attribute 'get_figure'

ฉันคาดว่ารอบชิงชนะเลิศoutput.pngจะมีอยู่และมีลักษณะเช่นนี้:

ฉันจะแก้ไขปัญหาได้อย่างไร

ลบget_figureและเพียงใช้sns_plot.savefig('output.png')

df = sns.load_dataset('iris')
sns_plot = sns.pairplot(df, hue='species', size=2.5)
sns_plot.savefig("output.png")

วิธีแก้ปัญหาที่แนะนำนั้นเข้ากันไม่ได้กับ Seaborn 0.8.1

ให้ข้อผิดพลาดต่อไปนี้เนื่องจากอินเทอร์เฟซ Seaborn มีการเปลี่ยนแปลง:

AttributeError: 'AxesSubplot' object has no attribute 'fig'
When trying to access the figure

AttributeError: 'AxesSubplot' object has no attribute 'savefig'
when trying to use the savefig directly as a function

การโทรต่อไปนี้ช่วยให้คุณสามารถเข้าถึงรูป (เข้ากันได้กับ Seaborn 0.8.1):

swarm_plot = sns.swarmplot(...)
fig = swarm_plot.get_figure()
fig.savefig(...) 

เท่าที่เห็นก่อนหน้านี้ในคำตอบนี้

ปรับปรุง: ฉันเพิ่งใช้วัตถุ PairGrid จากทะเลเพื่อสร้างพล็อตที่คล้ายกับหนึ่งในตัวอย่างนี้ ในกรณีนี้เนื่องจาก GridPlot ไม่ใช่วัตถุพล็อตเช่น sns.swarmplot จึงไม่มีฟังก์ชั่น get_figure () สามารถเข้าถึงรูป matplotlib ได้โดยตรง

fig = myGridPlotObject.fig

เหมือนที่แนะนำก่อนหน้านี้ในโพสต์อื่น ๆ ในกระทู้นี้

การแก้ปัญหาข้างต้นบางอย่างไม่ได้ผลสำหรับฉัน .figแอตทริบิวต์ไม่พบเมื่อฉันพยายามที่และผมก็ไม่สามารถที่จะใช้.savefig()โดยตรง อย่างไรก็ตามสิ่งที่ได้ผลคือ:

sns_plot.figure.savefig("output.png")

ฉันเป็นผู้ใช้ Python ที่ใหม่กว่าดังนั้นฉันไม่รู้ว่านี่เป็นเพราะการอัปเดตหรือไม่ ฉันอยากจะพูดถึงมันในกรณีที่คนอื่นพบปัญหาเดียวกันกับที่ฉันทำ

คุณควรจะสามารถใช้savefigวิธีการsns_plotได้โดยตรง

sns_plot.savefig("output.png")

เพื่อความชัดเจนกับรหัสของคุณหากคุณไม่ต้องการเข้าถึงรูป matplotlib ที่มีsns_plotอยู่ในนั้นคุณสามารถรับมันโดยตรงด้วย

fig = sns_plot.fig

ในกรณีนี้ไม่มีget_figureวิธีการที่เป็นรหัสของคุณถือว่า

ฉันใช้distplotและget_figureเพื่อบันทึกภาพสำเร็จ

sns_hist = sns.distplot(df_train['SalePrice'])
fig = sns_hist.get_figure()
fig.savefig('hist.png')

ลดจำนวนบรรทัดสำหรับผู้ค้นหา 2019:

import matplotlib.pyplot as plt
import seaborn as sns

df = sns.load_dataset('iris')
sns_plot = sns.pairplot(df, hue='species', height=2.5)
plt.savefig('output.png')

UPDATE หมายเหตุ: ก็เปลี่ยนไปsizeheight

มันใช้งานได้สำหรับฉัน

import seaborn as sns
import matplotlib.pyplot as plt
%matplotlib inline

sns.factorplot(x='holiday',data=data,kind='count',size=5,aspect=1)
plt.savefig('holiday-vs-count.png')

เป็นไปได้ที่จะสร้างfigureวัตถุmatplotlib จากนั้นใช้plt.savefig(...):

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

df = sns.load_dataset('iris')
plt.figure() # Push new figure on stack
sns_plot = sns.pairplot(df, hue='species', size=2.5)
plt.savefig('output.png') # Save that figure

คุณจะได้รับข้อผิดพลาดsns.figure.savefig("output.png")ในการใช้งานในท้องทะเล 0.8.1

ใช้แทน:

import seaborn as sns

df = sns.load_dataset('iris')
sns_plot = sns.pairplot(df, hue='species', size=2.5)
sns_plot.savefig("output.png")

เพียงแค่ FYI คำสั่งด้านล่างทำงานในทะเล 0.8.1 ดังนั้นฉันเดาว่าคำตอบเริ่มต้นยังคงใช้ได้

sns_plot = sns.pairplot(data, hue='species', size=3)
sns_plot.savefig("output.png")