データの可視化:散布図の作成

%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']を指定

コメント

タイトルとURLをコピーしました