2016-11-29 8 views
0

hereと同様のグラフを取得したいと思います。私のデータセットには、3つの機能と1つのバイナリターゲットがあります。目標= 0の場合、ドットは着色されるべきである。赤で。目標= 1の場合、ドットは着色されるべきである。青色の。scatter_matrixの作成

私はこのコードを実行すると、すべてのドットが同じ色持っているように、それが見えます:私は可能なグループ化を観察することができるように私は、目標値0と1のために異なる色を指定する方法を

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

を?

答えて

0

pandasplottingツールを使用してこのデータをプロットするには、今のようにcolor=['blue' if target == 1 else 'red']を引数として渡します。

しかし、代わりにseabornpairplot機能を使用することをおすすめします。これにより、チュートリアルのプロットに近いプロットが得られ、データポイントに基づいて色付けするには、hue='my_data_column'をパラメータとして指定するだけで済みます。 Cf. here

+0

「シーボーン」を使用していただきありがとうございます。このパッケージを使用してどのように行うかの例を挙げてください。私はそれを一度も使用していません。 – duckertito

+0

この場合、 'import seaborn as sns; –

+0

問題は、 'g = sns.pairplot(df [selected_features]、hue)を実行したときに、 = "target")g'の場合、プロットは表示されません(私はJupyterで作業しています)。生成されたように見えますが、表示されません。 'g.show'を実行するとエラーになります。 – duckertito