やりたいこと・やり方

pandasのDataFrameにおいて、列の要素や行ごとにfor文を回すように処理したいことがあると思います。
そんな時はapply()を使えば良いと分かりました。

想定する場面

5人の学生が数学と英語のテストを受け、その合計点で合否判定を出す。

ソースコード

main.py

import pandas as pd

def absent_result(x):
    """未受験の場合は30点とする"""

    if x == "未受験":
        return 30
    else:
        return x

def calculate_total_score(row):
    """数学と英語の合計点を出す"""

    return row['数学'] + row['英語']

def judge_pass_or_failure(x):
    """数学と英語の合計点が100点以上を合格、100点未満を不合格とする"""

    if x >= 100:
        return '合格'
    else:
        return '不合格'

if __name__ == "__main__":
    # データ準備
    name = ['田中', '伊藤', '鈴木', '加藤', '島田']
    math_score = [70, 38, 25, 40, '未受験']
    english_score = [90, 56, 44, 24, 89]

    df = pd.DataFrame({
        '名前': name,
        '数学': math_score,
        '英語': english_score
    })

    # 元データの出力
    print("### 元データ ###")
    print(df)
    print("")

    # 未受験者の処理
    df['数学'] = df['数学'].apply(absent_result)
    # 数学と英語の合計点の算出
    df['合計点'] = df.apply(calculate_total_score, axis=1)
    # 成績順に並び替え
    df = df.sort_values('合計点', ascending=False)
    # 合否判定
    df['判定'] = df['合計点'].apply(judge_pass_or_failure)

    print("### 最終結果 ###")
    print(df)

出力結果

### 元データ ###
   名前   数学  英語
0  田中   70  90
1  伊藤   38  56
2  鈴木   25  44
3  加藤   40  24
4  島田  未受験  89

### 最終結果 ###
   名前  数学  英語  合計点   判定
0  田中  70  90  160   合格
4  島田  30  89  119   合格
1  伊藤  38  56   94  不合格
2  鈴木  25  44   69  不合格
3  加藤  40  24   64  不合格

おわりに

4本目の記事になりました。最後までご覧いただきありがとうございました！
感想やアドバイス、何でもコメントお待ちしています。