Pandas.tseries.offsets.CustomBusinessMonthBeginを使いこなす

pandas.tseries.offsets.CustomBusinessMonthBegin.rollback は、Pandasの「Data offsets」機能で、カスタムビジネス月始のオフセットを指定された日付から過去方向にロールバックするために使用されます。

機能

この関数は、以下の機能を提供します。

• 指定された日付がカスタムビジネス月始でない場合、過去方向に最も近いカスタムビジネス月始の日付にロールバックします。
• ロールバック処理は、祝日や週末などの非営業日も考慮されます。
• 特定の曜日のみ営業日とするなど、複雑なカスタムビジネスカレンダーにも対応できます。

引数

CustomBusinessMonthBegin.rollback は以下の引数を受け取ります。

• date: ロールバック対象の日付。pandas.Timestamp 型または文字列型で指定できます。
• holidays: 祝日のリスト。pandas.DateOffset 型のオフセットのリストで指定できます。
• weekmask: 曜日のマスク。営業日を True 、非営業日を False で指定します。

戻り値

CustomBusinessMonthBegin.rollback は、ロールバック後の日付を pandas.Timestamp 型で返します。

使用例

import pandas as pd

# カスタムビジネスカレンダーの設定
holidays = [pd.offsets.DateOffset(days=1), pd.offsets.DateOffset(days=7)]
weekmask = "Mon Tue Wed Thu Fri"

# ロールバック対象の日付
date = pd.to_datetime("2024-04-05")

# ロールバック後の日付を取得
rolled_date = pd.tseries.offsets.CustomBusinessMonthBegin(holidays=holidays, weekmask=weekmask).rollback(date)

print(rolled_date)

この例では、2024年4月5日をロールバックします。4月5日は土曜日で非営業日なので、過去方向に最も近いカスタムビジネス月始である2024年4月1日が返されます。

補足

• CustomBusinessMonthBegin.rollback は、CustomBusinessMonthEnd と同様に、月始ではなく月末を基準としたロールバックにも使用できます。
• より複雑なオフセット設定には、CustomBusinessDay や Relativedelta などの他のオフセットクラスを使用できます。

Pandas.tseries.offsets.CustomBusinessMonthBegin.rollback サンプルコード

import pandas as pd

# 祝日と週末の設定
holidays = [pd.offsets.DateOffset(days=1), pd.offsets.DateOffset(days=7)]
weekmask = "Mon Tue Wed Thu Fri"

# ロールバック対象の日付
date = pd.to_datetime("2024-04-05")

# ロールバック後の日付を取得
rolled_date = pd.tseries.offsets.CustomBusinessMonthBegin(holidays=holidays, weekmask=weekmask).rollback(date)

print(f"ロールバック対象の日付: {date}")
print(f"ロールバック後の日付: {rolled_date}")

出力例:

ロールバック対象の日付: 2024-04-05
ロールバック後の日付: 2024-04-01

特定の曜日のみ営業日とする場合

import pandas as pd

# 祝日と週末の設定
holidays = [pd.offsets.DateOffset(days=1)]
weekmask = "Mon Wed Fri"

# ロールバック対象の日付
date = pd.to_datetime("2024-04-04")

# ロールバック後の日付を取得
rolled_date = pd.tseries.offsets.CustomBusinessMonthBegin(holidays=holidays, weekmask=weekmask).rollback(date)

print(f"ロールバック対象の日付: {date}")
print(f"ロールバック後の日付: {rolled_date}")

出力例:

ロールバック対象の日付: 2024-04-04
ロールバック後の日付: 2024-04-01

月末を基準としたロールバック

import pandas as pd

# 祝日と週末の設定
holidays = [pd.offsets.DateOffset(days=1), pd.offsets.DateOffset(days=7)]
weekmask = "Mon Tue Wed Thu Fri"

# ロールバック対象の日付
date = pd.to_datetime("2024-04-30")

# 月末を基準としたオフセット設定
offset = pd.tseries.offsets.CustomBusinessMonthEnd(holidays=holidays, weekmask=weekmask)

# ロールバック後の日付を取得
rolled_date = offset.rollback(date)

print(f"ロールバック対象の日付: {date}")
print(f"ロールバック後の日付: {rolled_date}")

出力例:

ロールバック対象の日付: 2024-04-30
ロールバック後の日付: 2024-04-01

複数月のロールバック

import pandas as pd

# 祝日と週末の設定
holidays = [pd.offsets.DateOffset(days=1), pd.offsets.DateOffset(days=7)]
weekmask = "Mon Tue Wed Thu Fri"

# ロールバック対象の日付
date = pd.to_datetime("2024-05-15")

# 2ヶ月分のロールバック
offset = 2 * pd.tseries.offsets.CustomBusinessMonthBegin(holidays=holidays, weekmask=weekmask)

# ロールバック後の日付を取得
rolled_date = offset.rollback(date)

print(f"ロールバック対象の日付: {date}")
print(f"ロールバック後の日付: {rolled_date}")

出力例:

ロールバック対象の日付: 2024-05-15
ロールバック後の日付: 2024-03-01

営業日数のカウント

import pandas as pd

# 祝日と週末の設定
holidays = [pd.offsets.DateOffset(days=1), pd.offsets.DateOffset(days=7)]
weekmask = "Mon Tue Wed Thu Fri"

# 開始日と終了日
start_date = pd.to_datetime("2024-03-01")
end_date = pd.to_datetime("2024-05-15")

# 営業日数のカウント
offset = pd.tseries.offsets.CustomBusinessMonthBegin(holidays=holidays, weekmask=weekmask)
business_days = offset.apply(start_date, end_date)

print(f"営業日数:

Pandasでカスタムビジネス月始を扱うその他の方法

pandas.offsets.DateOffset と for ループ

import pandas as pd

# 祝日と週末の設定
holidays = [pd.offsets.DateOffset(days=1), pd.offsets.DateOffset(days=7)]
weekmask = "Mon Tue Wed Thu Fri"

# ロールバック対象の日付
date = pd.to_datetime("2024-04-05")

# カスタムビジネス月始のオフセット設定
offset = pd.offsets.DateOffset(months=1)

# 過去方向にループ処理を行い、最初の日付を取得
for _ in range(12):
    rolled_date = date - offset
    if rolled_date.weekday() in weekmask and rolled_date not in holidays:
        break

print(f"ロールバック対象の日付: {date}")
print(f"ロールバック後の日付: {rolled_date}")

**2. numpy.vectorize と calendar モジュール

import numpy as np
import calendar

# 祝日と週末の設定
holidays = [pd.to_datetime("2024-01-01"), pd.to_datetime("2024-04-01")]
weekmask = "Mon Tue Wed Thu Fri"

# ロールバック対象の日付
date = pd.to_datetime("2024-04-05")

# カスタムビジネス月始判定関数
def is_business_day(date):
    return date.weekday() in weekmask and date not in holidays

# 過去方向に12ヶ月分の判定結果を取得
is_business_days = np.vectorize(is_business_day)(np.arange(date - pd.offsets.DateOffset(months=12), date))

# 最初の営業日を取得
rolled_date = date - pd.offsets.DateOffset(months=np.argmax(is_business_days[::-1]))

print(f"ロールバック対象の日付: {date}")
print(f"ロールバック後の日付: {rolled_date}")

自作関数

上記の方法に加えて、より複雑なカスタムビジネスカレンダーを扱う場合は、自作関数を用いる方法も考えられます。

Pandasでカスタムビジネス月始を扱うには、CustomBusinessMonthBegin 以外にもいくつかの方法があります。

それぞれの特徴を理解し、目的に合った方法を選択することが重要です。