python 条件分岐がうまくいかない

Question

Pythonの条件分岐ifで下記のコードを書いて実行したのですが、
すべて４が返されてしまいます。
変数meal_idには、５１種類の数値が割り振られているので、
それを５つに分類したいと思っています。

# meal_idを５つに分類　

def meal_cluster(x):
  if x["meal_id"] == "1558"or"2581"or"2707":
    res = 1
  if x["meal_id"] == "1778"or"1962"or"1993": 
    res = 2 
  if x["meal_id"] == "1109"or"2290"or"2664": 
    res = 3 
  if x["meal_id"] == "1885"or"2539": 
    res = 4
  else:
    res = 5
  return res

meal["meal_cluster"] = meal.apply(meal_cluster, axis=1)
meal.head()

下記の画像のとおり、なぜか、４が付けられてしまいます。
何かコードが間違えているのでしょうか。
教えていただけますでしょうか。

Comments

[ppaul]

全部１になるはずです。
4になるのは変ですね。
いや、4であってますね。

Answer 1

条件がいっぱいあるとメンドーですよね。
例えばこんな感じにしたらどうでしょうか?

def meal_cluster(x):
  meals = {
    "1558":1, "2581":1, "2707":1,
    "1778":2, "1962":2, "1993":2,
    "1109":3, "2290":3, "2664":3,
    "1885":4, "2539":4}
  if x["meal_id"] in meals:
    res = meals[x["meal_id"]]
  else:
    res = 5
  return res

Comments

[Shin_go]

丁寧にありがとうございます、大変ありがたいです。
ただ、すべて５になってしまいました。
DFに問題があるのかもしれません。

[Shin_go]

ダブルクォーテーションを外したら、意図通りに実行されました。本当にありがとうございました。

[takasima20]

(･∀･)b

Answer 2

ifの使い方と条件式が良くないです。

# meal_idを５つに分類　

def meal_cluster(x):
  if x["meal_id"] == "1558" or x["meal_id"] == "2581" or x["meal_id"] == "2707":
    res = 1
  elif x["meal_id"] == "1778" or x["meal_id"] == "1962" or x["meal_id"] == "1993": 
    res = 2 
  elif x["meal_id"] == "1109" or x["meal_id"] == "2290" or x["meal_id"] == "2664": 
    res = 3 
  elif x["meal_id"] == "1885" or x["meal_id"] == "2539": 
    res = 4
  else:
    res = 5
  return res

meal["meal_cluster"] = meal.apply(meal_cluster, axis=1)
meal.head()

条件式を修正した場合でも、elifを使用しないとresが4と5のみになってしまいます。
ifの使い方と条件式の書き方を学びましょう。

Pythonのif文による条件分岐の書き方

---

【追記】
手元で試した結果を追記します。こちらでは問題なさそうです。

import pandas as pd

meal = pd.DataFrame({'meal_id':['1558','1778','1109','1885','1000','2581','1962','2290','2539','2707','1993','2664']})
print(meal)
#	meal_id
#0	1558
#1	1778
#2	1109
#3	1885
#4	1000
#5	2581
#6	1962
#7	2290
#8	2539
#9	2707
#10	1993
#11	2664

def meal_cluster(x):
  if x["meal_id"] == "1558" or x["meal_id"] == "2581" or x["meal_id"] == "2707":
    res = 1
  elif x["meal_id"] == "1778" or x["meal_id"] == "1962" or x["meal_id"] == "1993": 
    res = 2 
  elif x["meal_id"] == "1109" or x["meal_id"] == "2290" or x["meal_id"] == "2664": 
    res = 3 
  elif x["meal_id"] == "1885" or x["meal_id"] == "2539": 
    res = 4
  else:
    res = 5
  return res

meal["meal_cluster"] = meal.apply(meal_cluster, axis=1)
print(meal)
#	meal_id	meal_cluster
#0	1558	1
#1	1778	2
#2	1109	3
#3	1885	4
#4	1000	5
#5	2581	1
#6	1962	2
#7	2290	3
#8	2539	4
#9	2707	1
#10	1993	2
#11	2664	3

Comments

[Shin_go]

丁寧に回答いただきありがとうございます。
大変ありがたいです。
ただ、今後は、すべて５になってしまいました。
elseまでの条件が、すべてfalseになってしまったということですよね。
難しいですね。。

[Shin_go]

何度もありがとうございます。
やはりすべて５になってしまいました。
DFに問題があるのかもしれませんね。

[Shin_go]

ダブルクォーテーションを外したら、意図通りに実行されました。本当にありがとうございました。

Answer 3

  if x["meal_id"] == "1885"or"2539": 
    res = 4

これは

  if (x["meal_id"] == "1885") or "2539": 
    res = 4

と同じですので、Falseではない"2539"は常に成り立ち、ブール値としてはTrueと同じ扱いになるからです。

公式ドキュメント 6.11. ブール演算 (boolean operation)をお読みください。

Comments

[Shin_go]

１、２，３，４，５に分類したいのですが、
例えば、１５５８の場合は、１になるはずなのに、４になってしまいますので。