将 Sci-Kit Learn Matplotlib 示例写入文件

Writing Sci-Kit Learn Matplotlib Example to File

我正在尝试测试我在 sci-kit 学习教程中找到的示例,但对其进行了修改,以便我可以将生成的图表写入文件。根据文档,写入文件的最佳方式是:

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

然而,在我看来,至少在 f.write() 中没有明确的对象可以作为论据。解决此问题的最佳方法是什么?唯一的选择是简单地找到一种方法来包装所有单独的绘图功能(例如 plt.xlabel("Sepal Length"))并引用它吗?代码见下方。

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

不确定你在问什么,但据我了解,请尝试

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