pandasデータフレームから最新の売上高と任意のn年前のデータを抽出したい。

Question

https://teratail.com/questions/223464
↑この投稿に関連した質問となります。

以下◆CSVのデータをPythonで、最新年の売り上げ高と任意のn年前のデータを使ってCAGRを計算したいと考えています。
先に示させていただいたURLにおいて、maxとminを使った計算方法が示されているところですが、
この抽出の仕方を、最新の年度と、n年前の年度の売上高を抽出する方法を教えてください。

参考に以下に自身の中途半端（頓挫した状態）のPythonコードを記載します。
このコードは、年度の並び替えをしてから、groupbyをして、そのあとgroup毎の一番上の値と、n番目の値を取り出そうと考えたものですが、【そのあと】以降の部分がわからなくなってしまった状態です。
また過去の投稿の回答のコードも掲載します。

◆CSV（この中には他の企業のデータも大量に含まれます。）
EDINET 証券コード　純資産額 　総資産額 売上高　　　 年度
E04846 97020 8755704000 13505695000 24434307000 2020
E04846 97020 6687528000 12243296000 22520258000 2019
E04846 97020 5556074000 7802415000 13395184000 2016
E04846 97020 5805187000 9195875000 16668195000 2017
E04846 97020 6185837000 9981400000 17441441000 2018

#私の中途半端なコードです。
def make_indicator_multiple():
    indicator_list =[]

    df = pd.read_csv(任意のパス,encoding="cp932")
    df = df.sort_values("年度",ascending=False)
    group = df.groupby("証券コード")

#過去質問の回答のコードです。
import io
import pandas as pd

dummy_data = """
会社名,期末,売上高
A,20190331,103
A,20180331,102
A,20170331,101
B,20130331,15
B,20110331,11
B,20120331,12
C,20150531,1000
C,20140531,100
C,20130531,10000
C,20120531,10
"""

df = pd.read_csv(io.StringIO(dummy_data), parse_dates=["期末"])
group = df.groupby("会社名")
min_idx = group["期末"].idxmin()
max_idx = group["期末"].idxmax()
min_data = df.iloc[min_idx].set_index("会社名")
max_data = df.iloc[max_idx].set_index("会社名")
ns = max_data["期末"].dt.year - min_data["期末"].dt.year
result = (max_data["売上高"] / min_data["売上高"]) ** (1 / (ns - 1)) - 1
print(result)
""" =>
会社名
A    0.019802
B    0.363636
C    9.000000
dtype: float64
"""

Answer 1

最新の年度と、n年前の年度の売上高を抽出する方法を教えてください。

こういうときは、最新から何年前かを表す列を作ることから考えます。

最新年度はgroupby('証券コード')['年度'].max()で求められますが、transformを使うことで、
もとのデータフレームの行の形のままで求めることができます。

df['n年前'] = df.groupby('証券コード')['年度'].transform('max') - df['年度']

このままでも計算できると思いますが、これをpivotすると使いやすいです。

df2 = df.pivot(index='証券コード', columns='n年前', values='売上高')

n = 2
print((df2[0] / df2[n]) ** (1 / (n - 1)) - 1)

Comments

[Oisan]

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

すごくスッキリしててかつ、すごくわかりやすい形になって感動しています。

ほんとうにありがとうございました！