python ifとapplyメソッドの使い方

kokyaku_new という顧客リストのデータフレームのuser_categoryというカラムに、
F1,F2,F3,M1,M2,M3
という情報が入っています。
F1,F2,F3、に０
M1,M2,M3、に１
というように性別分けができるような情報を追加したいと思っています。
下記のコードを実行すると、すべて１になってしまいます。
なにが間違えているかわかりますでしょうか？？

python
1
2# 顧客とに性別を付与
3def add_gender(x):
4  if ["user_category"]== ["F1"]:
5    res=0
6  elif ["user_category"]== ["F2"]: 
7    res=0 
8  elif ["user_category"]== ["F3"]: 
9    res=0 
10  else:
11    res=1
12  return res
13
14kokyaku_new["gender"] = kokyaku_new.apply(add_gender, axis=1)
15kokyaku_new
16

行動規範の内容に同意します

回答1件

ベストアンサー

こういう場合は、以下のように書きます。

python
1def add_gender(x):
2  if x == ["F1"]:
3    res=0
4  elif x == ["F2"]: 
5    res=0 
6  elif x == ["F3"]: 
7    res=0 
8  else:
9    res=1
10  return res
11
12kokyaku_new["gender"] = kokyaku_new["user_category"].apply(add_gender)