データフレームに行番号を追加させたい

Question

下のようなデータフレームで、インデックスは日付で保持したままその日付の行が何番目なのかを把握するための番号の列（2019-11-01をゼロとする）を追加させたいです。

例えば注釈(annotation)をつける際、引数には行番号を記入する必要があるかと思いますがその時に特定の日付が何番目なのかを知りたいからです。（そもそもannotationに直接日付を引数に渡すことはできるのでしょうか）

どなたかご返答いただけますと幸いです。

from pandas_datareader import data
import pandas as pd
import matplotlib.pyplot as plt
from datetime import datetime
%matplotlib inline

start = '2019-11-01'
end = '2020-11-01'

df = data.DataReader('^N225','yahoo', start, end)
df

Comments

[bsdfan]

日付から行番号を得たいだけなら、
df.index.get_loc(日付)
という方法もあります。

Answer 1

df = df.reset_index().reset_index().set_index('Date')

別解

df['index'] = df.reset_index().index

Comments

[o-fk]

ありがとうございます！

Answer 2

いくつか方法があります。

df.assign(index=range(len(df)))
# または
df.assign(idx=np.arange(len(df)))
# または
df.assign(idx=pd.RangeIndex(len(df)))

---

速度計測

import benchit
import numpy as np
import pandas as pd


def by_reset_reset(df):
    return df.reset_index().reset_index().set_index('index')


def by_reset(df):
    return df.assign(idx=df.reset_index().index)


def by_range(df):
    return df.assign(idx=range(len(df)))


def by_np_arange(df):
    return df.assign(idx=np.arange(len(df)))


def by_rangeindex(df):
    return df.assign(idx=pd.RangeIndex(len(df)))


df = pd._testing.makeDataFrame()
t = benchit.timings([by_reset_reset, by_reset, by_range, by_np_arange, by_rangeindex],
                    [df.sample(n, replace=True, random_state=0)
                     for n in 10**np.arange(6)])
t.plot(logx=True, logy=True, figsize=(10, 6))