インデックス内のすべての要素が条件を満たしているかの判定

Question

例えば店舗ID毎のすべての顧客カテゴリに"優良"と含まれていれば、
その店舗IDに"優良店舗フラグ"を立てるというコードを下記の様に記載致しました。

df002 = df001.set_index("店舗ID")

for row in df002.index[:]:
    if ("優良" in df002.loc[row,"顧客カテゴリ"]).all():
        df002["優良店舗フラグ"] = 1
    else:
        df002["優良店舗フラグ"] = 0

上記のコードを実装したところ、下記のエラー文となり実行が不可でした

AttributeError: 'bool' object has no attribute 'all'

こちらのエラーを解消する方法、または要件を満たすコードをご教示頂けますでしょうか。
宜しくお願い致します。

Comments

[meg_]

質問のコードですと重複して店舗ID毎の処理が実行されますので、df.groupbyを使用されるのが良いと思います。

Answer 1

店舗IDでgroupbyして、顧客カテゴリーのSeriesに対して「すべての値が'優良'かどうか」でフラグ(True, False)を設定する。

import pandas as pd


df001 = pd.DataFrame([[1, "優良"], [1, "優良"],
                      [2, "普通"], [2, "優良"],
                      [3, "優良"], [3, "優良"], [3, "優良"],
                      [4, "優良"],
                      [5, "普通"], [5, "劣悪"], [5, "優良"]],
                     columns={'店舗ID', '顧客カテゴリ'})


def quality_check(s):
    for value in s:
        if value != '優良':
            return False
    return True


df002 = pd.DataFrame(df001.groupby("店舗ID")["顧客カテゴリ"].apply(quality_check))
df002.rename(columns={'顧客カテゴリ': '優良店舗フラグ'}, inplace=True)
print(df002)

      優良店舗フラグ
店舗ID
1        True
2       False
3        True
4        True
5       False