Python if構文の書き方について

Question

前提・実現したいこと

日々の株価の、各移動平均線の上下の位置関係を判別するコードをPythonで書きたいです。
前提としては、各日ごとの株価データはあり、それをもとに移動平均の数値を計算しています。その数値（計算式）を使ってif文で大小を比較し、それをデータフレームに反映させたいと考えているのですが、うまく表現できずに困っています。

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

if文がうまく書けません。

該当のソースコード

# 指数平滑移動平均計算
def calc_ema(prices, period):
    ema = np.zeros(len(prices))
    ema[:] = np.nan # NaN で初期化
    ema[period-1] = prices[:period].mean() # 最初だけ単純移動平均
    for d in range(period, len(prices)):
        ema[d] = (ema[d-1] * (period - 1) + prices[d] * 2) / (period + 1)
    return ema

df['ema5'] = calc_ema(df.close, 5)
df['ema20'] = calc_ema(df.close, 20)
df['ema40'] = calc_ema(df.close, 40)

この計算をもとに、
ema5 > ema20 > ema40なら「1」と返して、ema20 > ema5 > ema40なら「2」と返して…
などとしたいができない、という状況です。どなたか、シンプルに書ける方法を教えていただけると助かります。よろしくお願いします。

試したこと

（上記記載済み）

補足情報（FW/ツールのバージョンなど）

（特になし）

Comments

[tiitoi]

pandas の質問をする際はサンプルデータを提示してください

Answer 1

各データごとに処理を行うような場合、DataFrameのapplyメソッドを使用します。

pandasで要素、行、列に関数を適用するmap, applymap, apply | note.nkmk.me

import random
import pandas as pd

random.seed(0)

df = pd.DataFrame({"ema5": [random.randrange(10) for _ in range(10)],
                   "ema20":[random.randrange(10) for _ in range(10)],
                   "ema40":[random.randrange(10) for _ in range(10)]})

def f(s):
    if s["ema5"] > s["ema20"] > s["ema40"]:
        return 0
    elif s["ema5"] > s["ema20"] < s["ema40"]:
        return 1
    # あとはお好きに
    else:
        return 3
print(df)
df["result"] = df.apply(f, axis=1)
print(df)
""" =>
   ema20  ema40  ema5
0      9      9     6
1      3      2     6
2      8      4     0
3      2      1     4
4      4      1     8
5      2      5     7
6      1      7     6
7      9      8     4
8      4      1     7
9      8      5     5
   ema20  ema40  ema5  result
0      9      9     6       3
1      3      2     6       0
2      8      4     0       3
3      2      1     4       0
4      4      1     8       0
5      2      5     7       1
6      1      7     6       1
7      9      8     4       3
8      4      1     7       0
9      8      5     5       3
"""

Comments

[qsuke]

分かりやすく説明いただき、ありがとうございます！
いただいたものをもとに、自分のやりたいことに展開して形にすることができました。