カラムの中の要素ごとに条件を当てはめられるか

Question

#やりたいこと
顧客データと購買データを使って、
データ加工の練習をしています。
そこでdefで関数を作る中で
色々条件を設定したいのですが、
一つ疑問があります。

##疑問
deta frameのカラムの一つであるIDごとに
条件を設定できるか

###具体的に言うと
dsというdataframeを作りました。
内容は

　cst_id　yymm_entry　Main_store_code　KaiinType　KaiinStatus　Gold_Kaiin　DMFlag_c　MailFlag_c　...
0 3147023052075 201711 3001.0 Web会員 登録済 通常会員 DM拒否 メール可 ...
1 3147023052075 201711 3001.0 Web会員 登録済 通常会員 DM拒否 メール可 ...
みたいな感じです。

この中にSt_SalesChannelというのがあり、アウトレット店、百貨店、路面店などがあります。

ここで、
pandas-plyの要領で、
from pandas_ply import install_oly, X, sym_call
install_ply(pd)

def add_flag3(data_input):
if ds_inout.St_SalesChannel == '路面店':
output = 1 if df_input.cst_id........
みたいな感じで書きたいのですが、
設定したい条件が
「同じIDで最初の行のChannelが路面店、
そして二回目以降の購入（同一IDの行で2行目以降）、
かつ路面店での購入」というものです。
少し複雑なのですが、
こういう条件を if と elseの組み合わせで
実現できるのでしょうか？？？

Answer 1

def add_flag3(data_input):
    df = data_input[data.duplicated()]
    df = data_input[data_input[["id"] == df["id"]]
    
    count = df['St_SalesChannel'].value_counts()["路面店"]
    
    if count > 1:
        return df

多分こうだと思います

Comments

[Pablito]

ご回答ありがとうございます。

from pandas_ply import install_ply, X, sym_call
install_ply(pd)

def add_flag3(ds_input):
df = ds_input[ds.duplicated()]
df = ds_input[ds_input[['cst_id'] == ds['cst_id']]]

count = df['St_SalesChannel'].value_counts()['路面店']

if count > 1:
return ds
ds.ix[ : , 'flag3'] = ds.apply(add_flag3, axis=1)
頂いたアドバイスを参考にこのように書いてみたのですが、
以下のようなエラーが起きました。

IndexingError: ('Unalignable boolean Series provided as indexer (index of the boolean Series and of the indexed object do not match).', 'occurred at index 0')
理由は分かりますでしょうか？？？

[退会済みユーザー]

条件に当てはまったものを返しているのでdsの行数が合わずエラーになっているので
import numpy as np

input_data = add_flag3()
ds["flag3"] = np.nan
for i in list(input_data.index):
df.loc[i] = input_data[i]
こうすればできると思います

[Pablito]

ありがとうございます！

Answer 2

良く伝わらないのですがこんな感じでしょうかね。
最終行のprint文で「1回目のChannelが路面店で、なおかつ2回目以降に1度でもChannelが路面店の人」のcst_idを出力しています。
適宜改変してください。

true_customer_list = []
for one_cst_data in all_cst_data:
    first_flag = false
    for one_purchase_data in all_purchase_data:
        if one_cst_data.cst_id == one_purchase_data.cst_id:
            if one_purchase_data.St_SalesChannel == '路面店':
                if first_flag:
                    true_customer_list.append(one_cst_data.cst_id)
                    break
                first_flag = true
            elif not first_flag:
                break
print(true_customer_list)