散布矩阵中的多个数据

Question

是否可以向a添加多个数据pandas.tools.plotting.scatter_matrix并为每组数据指定颜色？

我想用一组数据显示带有数据点的散点图,比如绿色,另一组用红色表示相同的散布矩阵.同样适用于对角线上的密度图.我知道这可以通过使用matplotlib的scatter函数来实现,但这并没有给我一个散点矩阵.

大熊猫的文件很清楚.

Answer 1

简短的答案是确定散点图中每个点的颜色,将其作用到数组中并将其作为color参数传递.

例:

from pandas.tools.plotting import scatter_matrix
import pandas as pd
from sklearn import datasets

iris = datasets.load_iris()
iris_data = pd.DataFrame(data=iris['data'],columns=iris['feature_names'])
iris_data["target"] = iris['target']

color_wheel = {1: "#0392cf", 
               2: "#7bc043", 
               3: "#ee4035"}
colors = iris_data["target"].map(lambda x: color_wheel.get(x + 1))
ax = scatter_matrix(iris_data, color=colors, alpha=0.6, figsize=(15, 15), diagonal='hist')