编写Sci-Kit学习Matplotlib示例到文件

时间:2015-11-16 17:48:48

标签: python matplotlib scikit-learn

我正在尝试测试我在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()

1 个答案:

答案 0 :(得分:2)

不确定你在问什么,但根据我的理解,试试

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