Pandas scatter_matrix 绘图 - 附加参数

Question

我正在运行 Python 3.6 和 Pandas 版本 0.19.2。在下面的代码示例中，我对 Pandas 绘图函数scatter_matrix() 有两个问题：

**1.**如何在散点图中对Label 列的观察结果进行颜色标记？

**2.**如何指定对角线直方图的 bin 数量？我可以单独执行此操作还是只为所有人执行一个 bin 编号？

import pandas as pd
import numpy as np

N= 1000
df_feat = pd.DataFrame(np.random.randn(N, 4), columns=['A','B','C','D'])
df_label = pd.DataFrame(np.random.choice([0,1], N), columns=['Label'])
df = pd.concat([df_feat, df_label], axis=1)
axes = pd.tools.plotting.scatter_matrix(df, alpha=0.2)

这是链接到this more general 之一。

Answer 1

要回答你的第一个问题，可能有一种不那么“笨拙”的方式，但是

scatter_matrix(df,c=['r' if i == 1 else 'b' for i in df['Label']])

回答第二个：

散点矩阵可以使用pd.hist() api来使用字典中传递的hist关键字

scatter_matrix(df,hist_kwds={'bins':5})