我有一个Pandas dataframe,其中包含一些列,例如:
product color type
A blue 2
A red 3
A red 1
B red 1
B red 2
B green 5
C yellow 1
C yellow 2
C blue 3
.
.
.
我想创建一个看起来像决策树的方案,用我们看到的列和值显示所有可能性。就像这样:
A B C
blue red red green yellow blue
1 2 3 1 2 5 1 2 3
但是如果用箭头连接这些值,A将与蓝色和红色相连,这将与1,2和3相连,依此类推。就像我们可视化决策树的方式一样,但我不想创建一个统计模型,只显示可用的可能性,以及每种可能性的观察数量。
这可能吗?我搜索了所有类型的术语,但找不到任何接近的东西。
我想运行一个机器学习模型,然后绘图,但我不认为这会得到我想要的,因为它会通过信息增益来探索事物,我想展示一切。
1条答案
按热度按时间pxy2qtax1#
你可以使用
networkx
来构建一个有向图:图形(
graphviz
渲染后):用于绘图的代码(导出图像并以jupyter输出),请参阅here: