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

Question

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

# 顧客とに性別を付与
def add_gender(x):
  if ["user_category"]== ["F1"]:
    res=0
  elif ["user_category"]== ["F2"]: 
    res=0 
  elif ["user_category"]== ["F3"]: 
    res=0 
  else:
    res=1
  return res

kokyaku_new["gender"] = kokyaku_new.apply(add_gender, axis=1)
kokyaku_new

Answer 1

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

def add_gender(x):
  if x == ["F1"]:
    res=0
  elif x == ["F2"]: 
    res=0 
  elif x == ["F3"]: 
    res=0 
  else:
    res=1
  return res

kokyaku_new["gender"] = kokyaku_new["user_category"].apply(add_gender)

Comments

[Shin_go]

ありがとうございました。解決ししました！