【python】Dataframe

Question

実現したいこと

いつも大変お世話になっております。
下図のデータフレームの斤量と上りの散布図をcourse_lenごとにプロットしたいと考えています。1つの散布図に全てをプロットするのではなく、course_lenの数の分だけ表示させたいです。

発生している問題・エラーメッセージ

下のコードのようにplt.subplotで散布図をcourse_lenの分、用意することはできましたが、全てのデータが全ての散布図に入ってしまいます（コード下の画像参照）。

values = list(df['course_len'].unique())

fig = plt.figure(figsize=(10,20))

for i in range(len(values)):
    plt.subplot(len(values), 1, i+1)
    plt.title('course_len ' + str(values[i]))
    for value in values:
        x = df[df['course_len'] == value]['斤量']
        y = df[df['course_len'] == value]['上り']
        plt.scatter(df[df['course_len'] == value]['斤量'], df[df['course_len'] == value]['上り'])

これをcourse_lenごとに散布したいと考えています。ご教示のほどよろしくお願いいたします。

Answer 1

groupby を使ってみてもよいかもしれません。

import pandas as pd
import numpy as np
from matplotlib import pyplot as plt

N = 14
df = pd.DataFrame({
    'course_len': np.linspace(10, 36, N).repeat(N),
    '斤量': np.random.randint(0, 26, N*N) * 0.5 + 47.0,
    '上り': np.random.uniform(32, 38, N*N),
})

#
axes = plt.subplots(N, 1, figsize=(10, 20))[1].flatten()
for (c, g), ax in zip(df.groupby('course_len'), axes):
    g.plot.scatter(x='斤量', y='上り', ax=ax, title=f'course_len {c}')

plt.setp(axes, xlim=[46, 60], ylim=[32, 38], xlabel='', ylabel='')
plt.tight_layout()
plt.show()

Comments

[cheriraoka]

回答ありがとうございます。setp関数は初見でした。勉強させていただきます。

Answer 2

とりあえず以下のようにdf['course_len'] == values[i]という条件で抽出すべきで、二重にループする必要はありません。
なお、各グラフのX,Y範囲は全体が収まるような共通の値をセットすると、比較してみたい場合に分かりやすいかもです。

import pandas as pd
from io import StringIO
import matplotlib.pyplot as plt

s = """course_len,斤量,上り
10.0,48.0,33.40
10.0,48.5,35.90
36.0,54.0,36.25
36.0,56.0,36.00
36.0,57.0,35.50"""
df = pd.read_csv(StringIO(s))

values = list(df['course_len'].unique())

fig = plt.figure(figsize=(10,20))

for i in range(len(values)):
    plt.subplot(len(values), 1, i+1)
    plt.title('course_len ' + str(values[i]))
    x = df[df['course_len'] == values[i]]['斤量']
    y = df[df['course_len'] == values[i]]['上り']
    plt.scatter(x, y)

plt.show()

Comments

[cheriraoka]

回答ありがとうございます。valuesの要素を取り出せば良かったのですね。また、グラフの範囲についてもアドバイスありがとうございます。