データフレームに指定のインデックスが存在しない場合、別のインデックスを利用するには？

Question

指定した銘柄の株価を取得し、表示させる簡易的なプログラムを作っています。
指定した日付の株価のみを表示させたいと思っているのですが、問題があります。

指定した日付が土曜・日曜だった場合、そもそもデータが存在せず取得できません。（株式市場の仕組みはよくわかっていません）
そこで指定した日付が土曜・日曜だった場合（データフレームにインデックスが存在しない場合）、翌日か翌々日の株価（終値）を取得したいのですが、実現の方法がわからず困っています。
問題解決の方法・手段をお持ちの方がおられましたらご教示お願いいたします。

from pandas_datareader import data

# 任天堂の株価をデータフレームで取得
sec_code = 7974
df = data.DataReader(f'{sec_code}.JP', 'stooq')

# 株価を知りたい日付のリストを作成
days = ['2017-03-31', '2018-03-31', '2019-03-31', '2020-03-31', '2021-03-31']

# リスト化した日付の終値を表示
for i in days:
    price = df.loc[i, 'Close']
    print(price)

利用環境
Python 3.7.3 64-bit
Windows11 Pro 64bit

Answer 1

以下は first_valid_index() を使う方法です。

# リスト化した日付の終値を表示
dfx = df.sort_index()
for i in days:
    t = dfx.loc[i:, 'Close'].first_valid_index()
    print(f'{i} => {t.date()}')
    price = df.loc[t, 'Close']
    print(f'終値: {price}')

2017-03-31 => 2017-03-31
終値: 24727.84
2018-03-31 => 2018-04-02
終値: 46049.99
2019-03-31 => 2019-04-01
終値: 31313.36
2020-03-31 => 2020-03-31
終値: 41320.37
2021-03-31 => 2021-03-31
終値: 61810.0

Comments

[castlemist]

はじめて見るメソッドで勉強になります。
日本語の事例が少なく扱うのが厄介そうですが、
使いこなせればコードを短縮できそうですね。
ご回答どうもありがとうございました。助かりました。

Answer 2

以下のようにします。

>>> import pandas as pd
>>> from pandas_datareader import data
>>>
>>> df = data.DataReader(f'{sec_code}.JP', 'stooq')
>>>
>>> days = ['2017-03-31', '2018-03-31', '2019-03-31', '2020-03-31', '2021-03-31']
>>>
>>> df.loc[[min([d for d in df.index if d >= day]) for day in pd.to_datetime(pd.Series(days))]]['Close']
Date
2017-03-31    24727.84
2018-04-02    46049.99
2019-04-01    31313.36
2020-03-31    41320.37
2021-03-31    61810.00
Name: Close, dtype: float64

Comments

[castlemist]

ご回答どうもありがとうございます。
内包表記を読むのに時間がかかりましたが、
おかげさまで問題を解決することができました。
とても参考になりました。
どうもありがとうございました。

Answer 3

指定した日付が土曜・日曜だった場合、次の日を指定するようにして、そのうえで、指定した日付が土曜・日曜だった場合、次の日を指定するようにしましょう