将Sci-Kit学习Matplotlib示例写入文件

Question

I'm trying to test out an example I found in a sci-kit learn tutorial but modify it so that I can write the resulting charts to a file. According to the documentation the best way to write to a file is:

with open('path\file', 'w') as f:
        f.write()

However, to my eyes at least there isn't a clear object to place as an argument in f.write() . What is the best way to fix this? Is the only option to simply find a way to some how wrap up all the separate plot features (eg plt.xlabel("Sepal Length") ) and reference that? See below for the code.

from __future__ import division
from sklearn import datasets
import matplotlib.pyplot as plt
import seaborn as sns
#matplotlib inline
sns.set(style='ticks', palette='Set2')
import pandas as pd
import numpy as np
import math

data = datasets.load_iris()
X = data.data[:100, :2]
y = data.target[:100]
X_full = data.data[:100, :]

setosa = plt.scatter(X[:50,0], X[:50,1], c='b')
versicolor = plt.scatter(X[50:,0], X[50:,1], c='r')
plt.xlabel("Sepal Length")
plt.ylabel("Sepal Width")
plt.legend((setosa, versicolor), ("Setosa", "Versicolor"))
sns.despine()

with open('path\file', 'w') as f:
    f.write()

Answer 1

不确定您要问的是什么，但据我了解，请尝试

pylab.savefig('foo.png') pylab.savefig('foo.pdf')