【python】dataframeの数値を条件別に当てはめ，列を追加したい

Question

実現したいこと

vscodeを用いてpythonでdataframeの計算をしています．

dataframeの特定の列の数値を条件に当てはめ，新しい列に計算した値を代入したいと考えております．
具体的には，下記の表において，num1×5の値が
・10以下
・10100
・100200
・200~
の場合に，それぞれ違う数値を新しく作るnum3の列に代入したいです．

a.csv↓
||name1|num1|num2
|:--|:--:|--:|
|0|a|1|3
|1|b|2|4
|2|a|11|5
|3|b|12|6
|4|a|50|7
5|b|51|8|

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

if文の中でdataframeを用いると，dataframe全体で考えた際の真偽が判断できないためこのようなエラーが出ていると把握しました．
しかし，条件ごとに違う値を入れる方法がわからず，お手数をおかけしますがご教示いただけますと幸いです．

f"The truth value of a {type(self).__name__} is ambiguous. "
ValueError: The truth value of a Series is ambiguous. Use a.empty, a.bool(), a.item(), a.any() or a.all().

該当のソースコード

import pandas as pd

df = pd.read_csv('a.csv')
A = 5

print(df)

if 0 < A * df['num1'] < 10 :
    df['num3'] = 5
elif 10 <= A * df['num1'] < 100 :
    df['num3'] = 10 *A *df['num2']
elif 100 <= A * df['num1'] < 200 :
    df['num3'] = 15 *A *df['num2']
elif 200 <= A * df['num1'] :
    df['num3'] = 20 *A *df['num2']
else :
    df['num3'] = 'error'

print(df)

補足情報

pandasのバージョンは1.3.4です

追加

追加の質問で恐縮ですが，name1がaの行のみにこの値を代入することは可能でしょうか？
if文の中でdataframeを使わなかった際は，
df['num2'][df['name1'].str.contains('a')]*0+5
のように指定していたのですが，当たり前ですがこのままではうまく回らず…

Answer 1

import pandas as pd

df = pd.read_csv('a.csv')
A = 5

def translate(r):
    num1, num2 = r['num1'], r['num2']
    n = A * num1
    if 0 < n < 10:
        return 5
    elif 10 <= n < 100:
        return 10 * A * num2
    elif 100 <= n < 200:
        return 15 * A * num2
    elif 200 <= n:
        return 20 * A * num2
    else:
        return 'error'

df['num3'] = df.apply(translate, axis=1)

print(df.to_markdown(index=False))

name1	num1	num2	num3
a	1	3	5
b	2	4	200
a	11	5	250
b	12	6	300
a	50	7	700
b	51	8	800

Comments

[lmk]

丁寧にご教示いただきありがとうございます。
追記したものと重なってしまい恐縮なのですが，name1がaの行のみにこの値を代入することは可能でしょうか？

[melian]

はい、可能ですが、name1 が a 以外の場合の num3 はどうしますか？（例えば 0 を入れるなど）

[lmk]

解決することができました。お答えいただきありがとうございました！

Answer 2

以下のように.applyにて行毎に値を取得、判定して結果値を返すことができます。

import pandas as pd

df = pd.DataFrame({'name1':list('aaabbb'), 'num1':[1,11,50]*2, 'num2':[3,5,7]*2})
A = 5

def func(row):
    if 0 < A * row['num1'] < 10 :
        return 5
    elif 10 <= A * row['num1'] < 100 :
        return 10 *A *row['num2']

    return 'error'

df['num3'] = df[df['name1']=='a'].apply(func, axis=1)
#df['num3'] = 0  # NaNがいやなら必要に応じてデフォルト値を設定
print(df)
"""
  name1  num1  num2   num3
0     a     1     3      5
1     a    11     5    250
2     a    50     7  error
3     b     1     3    NaN
4     b    11     5    NaN
5     b    50     7    NaN
"""

Comments

[lmk]

丁寧にご教示いただきありがとうございます。
関数とapplyを組み合わせることで実行できるのですね、大変勉強になりました。
追加の質問で恐縮ですが，name1がaの行のみにこの値を代入することは可能でしょうか？
if文の中でdataframeを使わなかった際は，
df['num2'][df['name1'].str.contains('a')]*0+5
のように指定していたのですが，このままではうまく回らず…

[8524ba23]

回答を修正しました。
apply対象のdfをあらかじめ条件で絞ればよいです。

[lmk]

関数の前に絞ったらよいのですね…！
追加の質問にもかかわらずお答えいただきありがとうございました。