python 特定の値ごとにグループ化した新たなカラムを作りたい

Question

実現したいこと

特定の値ごとにグループ化した新たなカラムを作りたい

使用データ

df
#     user_id    number         
#0        47      9          
#1        47      9          
#2        48      8          
#3        49      7         
#4        49      7          

希望データ型

df
#     user_id    number   group     
#0        47      9        1  
#1        47      9        1 
#2        48      8        2  
#3        49      7        2
#4        49      7        2

条件:

• 9であれば、group 1
• 7 or 8であれば、group 2

試したこと

def conv_func(row):
    if row[1].isin([9]) :
        return 1
    elif row[1].isin([7,8]):
        return 2
    else:
        raise RuntimeError(str(row))


df["group"] = df.apply(conv_func)

→詳しい方、ご教授いただけたら、幸いです。
何卒よろしくお願いいたします。

Answer 1

conv = {9:1,8:2,7:2}
の様な変換用のdictを作成しておき
df['number'].map(conv)
のように'number'列に変換を適用するとよいのではないでしょうか。

import pandas as pd
df = pd.DataFrame({'user_id':[47,47,48,49,49],'number':[9,9,8,7,7]})
conv = {9:1,8:2,7:2}
df['group'] = df['number'].map(conv)
print(df)
#   user_id  number  group
#0       47       9      1
#1       47       9      1
#2       48       8      2
#3       49       7      2
#4       49       7      2

Answer 2

7、8、9しかないのであれば

df['group']= 2
df[df['number']==9]['group'] = 1

# queryを利用しても良い
df[df.query("number == 9")] = 1

7,8,9以外もある場合

df[df.query("number == 9")] = 1
df[df.query("number == 7 or number == 8")] = 2

# isinでも良い
df[df['number'].isin([7,8])] = 2

動かしていないので、間違いがあったらすいません。