python - 在 DecisionTree 中使用 StringIO 的目的是什么

Question

我正在写一个决策树，下面的代码是完整代码的一部分:

def show_tree(tree, features, path):
    f = io.StringIO()
    export_graphviz(tree, out_file=f, feature_names=features)
    pydotplus.graph_from_dot_data(f.getvalue()).write_png(path)
    img = misc.imread(path)
    plt.rcParams['figure.figsize'] = (20,20)
    plt.imshow(img)

谁能告诉我在这里使用 StringIO 的目的是什么？

Answer 1

Python 不是我的主要语言，但我认为您的问题的答案非常简单，不需要大量研究。

这里使用StringIO来维护输入/输出文本流。您的函数显示树，但是要做到这一点，它需要一种方法，某种数据传输高速公路。

此处 f = io.StringIO() 您正在初始化数据流。之后，您可以随意使用它，在这种特殊情况下:

1. export_graphviz(tree, out_file=f, feature_names=features)

此处:out_file=f 您使用 initialized before f = io.StringIO(); 将数据导出到您的流。由于 StringIO 是内存中的文本文件，您基本上将数据放在流对象中以供进一步使用。多亏了这一点，您不必将数据写入 .dot 文件，而是临时保存它。(临时意味着只要您的流正在使用中)

More about this particular case

1. pydotplus.graph_from_dot_data(f.getvalue()).write_png(path)

此处:f.getvalue() 您从 .dot 数据生成图表。在最基本的使用中，您应该确保指向 .dot 文件的路径，之前生成的数据将存储在该文件中，但您不必这样做!这就是诀窍，您的数据仍在您事先创建和填充的流对象中!所以现在您所要做的就是直接将其定向到此函数，该函数将使用该数据生成图形图像并将其保存为 .png 文件。

系统文件和您的程序之间的通信可以通过多种方式建立，但通常您使用流。您在一开始就初始化流，使用它然后关闭。每个 std::cout 或 std:err(我的主要语言引用，对那个非 python 示例感到抱歉)都是那个流。 Stream 允许您在您的程序和指定的 tagret(例如控制台，或在本例中为文件)之间保持数据交换，但是您也可以将它用作临时存储空间，在特定情况下这将加快图像生成过程，因为您不需要确实必须将数据写入文件并将其加载到文件中。得益于此，您所要做的就是按照其他函数可接受的顺序将数据写入流，然后使用完全相同的流读取该数据以生成图像。

More about StringIO