如何在Graphviz中显示功能名称？

Question

我正在Graphviz中构建一个树，我似乎无法显示要素名称，我已经定义了一个包含如下功能名称的列表：

names = list(df.columns.values)

打印哪些：

    ['Gender',
 'SuperStrength',
 'Mask',
 'Cape',
 'Tie',
 'Bald',
 'Pointy Ears',
 'Smokes']

因此正在创建列表，稍后我会像这样构建树：

 export_graphviz(tree, out_file=ddata, filled=True, rounded=True, special_characters=False, impurity=False, feature_names=names)

但是最终图像仍然具有列为X []的功能名称：

如何显示实际的功能名称？ （Cape而不是X [3]等）

Answer 1

我只能想象这与将名称作为值数组传递有关。如果直接传递列，它可以正常工作：

export_graphviz(tree, out_file=ddata, filled=True, rounded=True, special_characters=False, impurity=False, feature_names=df.columns)

如果需要，您还可以对列进行切片：

export_graphviz(tree, out_file=ddata, filled=True, rounded=True, special_characters=False, impurity=False, feature_names=df.columns[5:])