python始めたての初心者です。
独学でやっていて、Kaggleにたどり着いたので試しにやってみようと思ったところ、次の疑問ができました。
以下のような.csvファイルを配列として読み込み、散布図を書くことを考えています。
横軸にValue1、縦軸にValue2をとってプロットするのですが、BoolがTrueの場合のプロットだけ色を変えるなどで強調したいです。
どんな方法で記述すればよいでしょうか。
ID, Value1, Value2, Sex, Bool
0, 2135, 54, male, True
1, 1133, 28, male, False
2, 5430, 30, female, True
...
現状自分で記述したコードは以下のようになっています。
import pandas as pd
import matplotlib.pyplot as plt
...
test = pd.read_csv("train.csv")
plt.scatter(test["Value1"],test["Value2"])
plt.show()
初めは、testの{Bool}ラベルの値によって、if文で一つずつプロットすることを考えていたのですが、
変数の型をtype()関数で調べると意味不明なものになっていて？？？でした。そこから先に進めていません。
scatter()の引数cに、色を指定したx/yと同じ長さの配列を渡すとその通りプロットされます。
plt.scatter([0, 1, 2, 3], [0, 1, 2, 3], c=["red", "blue", "green", "black"])
したがって、np.where()関数でTrue/Falseをそれぞれ別の値に置換することで目的の結果になります。
import numpy as np
c_array = np.where(test["Bool"], "Trueの色", "Falseの色")
plt.scatter(test["Value1"], test["Value2"], c=c_array)
@BenBecky
Questioner私だったらseaborn使って描画します。
import pandas as pd
import seaborn as sns
test = pd.DataFrame({
'Value1': [2135, 1133, 5430],
'Value2': [54, 28, 30],
'Bool': [True, False, True]
})
sns.scatterplot(data=test, x='Value1', y='Value2', hue='Bool')
今回に限らずデータの可視化にseabornは便利なので調べてみてもいいかもです！
@BenBecky
Questioner