创建scatter_matrix

时间:2016-11-29 17:19:56

标签: python pandas

我希望得到一张类似于已发布的here的图表。在我的数据集中,我有3个功能和1个二进制目标。当目标= 0时,那些点应该是彩色的,例如红色的。当目标= 1时,则点应该是彩色的,例如蓝色。

当我运行此代码时,看起来所有点都具有相同的颜色:

from pandas.tools.plotting import scatter_matrix
scatter_matrix(df, alpha=0.2, figsize=(6, 6), diagonal='kde')

如何为目标值0和1指定不同的颜色,以便我可以观察可能的分组?

1 个答案:

答案 0 :(得分:0)

要像现在一样使用pandas plotting工具绘制此数据,请将color=['blue' if target == 1 else 'red']作为参数传递。

但是,我建议使用seaborn pairplot工具。这将导致绘图更接近教程中的绘图,并且基于数据点进行着色,您只需将hue='my_data_column'指定为参数。 Cf. here