【Python_Pandas】同施設に属する営業に人数分値を按分する方法をご教示ください

Question

上記の様に登録されたデータの"合計売上"に"Aシェア"、"Bシェア"を下記のデータに按分したいと考えております。

按分イメージは下記です

こちらを実現するpythonコード例をご教示頂きますと幸いです。
(施設毎に、割る数字が変化するのでそれをどのように実現するのかをお聞きしたいです)

宜しくお願い致します。

Answer 1

手順

施設毎に、割る数字が変化するのでそれをどのように実現するのかをお聞きしたいです

• 1. DataFrame.merge() で2つの DataFrame を "施設コード" でマージする。

施設コード	営業コード	合計売上	Aシェア	Bシェア
1001	2091	400	0.4	0.1
1001	3093	400	0.4	0.1
1001	6095	400	0.4	0.1
1001	2093	400	0.4	0.1
1002	2092	500	0.2	0.2
1002	5405	500	0.2	0.2
1002	4302	500	0.2	0.2
1002	4309	500	0.2	0.2
1002	4322	500	0.2	0.2
1003	5498	1000	0.3	0.5
1003	4392	1000	0.3	0.5
1004	3942	600	0.4	0.4
1005	3984	400	0.3	0.4

• 1. DataFrame.groupby() で "施設コード" で集計する。
• 1. GroupBy.apply() で各グループごとに "合計売上", "Aシェア", "Bシェア" の列をそのグループの数で除算する関数を作り、指定する。

コード

import pandas as pd

codes = pd.read_csv("codes.csv")
sales = pd.read_csv("sales.csv")

# "施設コード" で結合する。
merged = codes.merge(sales, left_on="施設コード", right_on="施設コード", how="outer")

def func(g):
    cols = ["合計売上", "Aシェア", "Bシェア"]  # 営業コード数で除算する列
    g[cols] = g[cols].div(len(g.index))
    return g

ret = merged.groupby("施設コード").apply(func)
ret.to_csv("output.csv")

入出力

入力ファイル

営業コードが記載された CSV (codes.csv)

施設コード,営業コード
1001,2091
1001,3093
1001,6095
1001,2093
1002,2092
1002,5405
1002,4302
1002,4309
1002,4322
1003,5498
1003,4392
1004,3942
1005,3984

売上が記載された CSV (sales.csv)

施設コード,合計売上,Aシェア,Bシェア
1001,400,0.4,0.1
1002,500,0.2,0.2
1003,1000,0.3,0.5
1004,600,0.4,0.4
1005,400,0.3,0.4

出力 CSV (output.csv)

施設コード,営業コード,合計売上,Aシェア,Bシェア
1001,2091,100.0,0.1,0.025
1001,3093,100.0,0.1,0.025
1001,6095,100.0,0.1,0.025
1001,2093,100.0,0.1,0.025
1002,2092,100.0,0.04,0.04
1002,5405,100.0,0.04,0.04
1002,4302,100.0,0.04,0.04
1002,4309,100.0,0.04,0.04
1002,4322,100.0,0.04,0.04
1003,5498,500.0,0.15,0.25
1003,4392,500.0,0.15,0.25
1004,3942,600.0,0.4,0.4
1005,3984,400.0,0.3,0.4

Comments

[D7U]

大変ご丁寧なご回答誠にありがとうございます！
解決いたしました。
Group byは単純に指定カラムで集計するのみの機能かと思っておりましたが、
applyと組み合わせることで、"〇〇毎に(今回であれば、施設毎に)関数を適用する"という使い方もできると初めて認識致しました。