我希望得到一张类似于已发布的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指定不同的颜色,以便我可以观察可能的分组?
答案 0 :(得分:0)
要像现在一样使用pandas
plotting
工具绘制此数据,请将color=['blue' if target == 1 else 'red']
作为参数传递。
但是,我建议使用seaborn
pairplot
工具。这将导致绘图更接近教程中的绘图,并且基于数据点进行着色,您只需将hue='my_data_column'
指定为参数。 Cf. here