scatter_matrix的创建

Question

我想要一个类似于阴阳这里的图表。在我的数据集中，我有3个特性和1个二进制目标。当target=0时，那么点应该被着色，例如用红色。当target=1时，那么点应该被着色，例如蓝色。

当我运行这个代码时，看起来所有的点都有相同的颜色：

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

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

Answer 1

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

但是，我建议改用seaborn pairplot工具。这将导致一个更接近于本教程中的绘图，并且基于数据点进行着色，您只需要指定hue='my_data_column'作为参数。请参阅这里。