%matplotlib inline
import pandas as pd
import matplotlib.pyplot as plt
df = pd.read_csv('./input/jh_heights_weights.csv')
men = df[df["y"] == 1] y=1を男性と指定
women = df[df["y"] == 0]
plt.scatter(men['height'], men['weight'], color='green')
plt.scatter(women['height'], women['weight'], color='blue');
plt.scatter
を使って散布図をプロット
- 第1引数にデータの横軸の値、第2引数にデータの縦軸の値を渡します
men = df[df['y'] == 1]
のように事前に男女のデータを分割しています- 横軸に
men['height]
、縦軸にmen['weight']
を指定
コメント