Pandasのset_closed()でIntervalArrayを両側閉区間「both」に変換する方法

Pandasのset_closed()メソッドを使用すると、現在のIntervalArrayと同一の内容を持ちながら、指定した側で閉じた（closed）新しいIntervalArrayを返すことができます。左右両方の境界を含む閉区間にしたい場合は、パラメータに"both"を指定します。これにより、各区間が [0, 1] のように両端を含む形式へと変換されます。

必要なライブラリのインポート

まず、必要なライブラリをインポートします。

import pandas as pd

IntervalArrayの作成

from_breaks()メソッドを使って、0から5までの境界値からIntervalArrayを作成します。

index = pd.arrays.IntervalArray.from_breaks(range(6))

区間の内容を確認する

作成したIntervalArrayを表示して、中身を確認してみましょう。

print("IntervalIndex...\n", index)

set_closed()で"both"を指定する

現在の配列と同一でありながら、指定した側（ここでは"both"＝両側）で閉じられたIntervalArrayを返します。

print("\nResult...", index.set_closed('both'))

完全なコード例

以下に、ここまでの手順をまとめたサンプルコードを示します。

import pandas as pd

# IntervalArrayを作成
index = pd.arrays.IntervalArray.from_breaks(range(6))

# 区間を表示
print("IntervalIndex...\n", index)

# 各区間の長さを表示
print("\nIntervalIndex length...\n", index.length)

# 左側の境界値を表示
print("\nThe left bound for the IntervalIndex...\n", index.left)

# 右側の境界値を表示
print("\nThe right bound for the IntervalIndex...\n", index.right)

# 現在の配列と同一で、指定した側（"both"）で閉じられた
# IntervalArrayを返す
print("\nResult...", index.set_closed('both'))

出力結果

上記のコードを実行すると、次のような出力が得られます。

IntervalIndex...
<IntervalArray>
[(0, 1], (1, 2], (2, 3], (3, 4], (4, 5]]
Length: 5, dtype: interval[int64, right]

IntervalIndex length...
Int64Index([1, 1, 1, 1, 1], dtype='int64')

The left bound for the IntervalIndex...
Int64Index([0, 1, 2, 3, 4], dtype='int64')

The right bound for the IntervalIndex...
Int64Index([1, 2, 3, 4, 5], dtype='int64')

Result... <IntervalArray>
[[0, 1], [1, 2], [2, 3], [3, 4], [4, 5]]
Length: 5, dtype: interval[int64, both]

ポイントの解説

元のIntervalArrayは右側のみが閉じている（right）状態ですが、set_closed('both')を実行することで、[0, 1]のように左端・右端の両方が含まれる閉区間へと変換されているのが分かります。なお、set_closed()は元の配列を変更せず、新しいIntervalArrayを返す点にも注意してください。指定できる値は 'left'、'right'、'both'、'neither' の4種類です。