Python Pandas入門：from_breaks()でIntervalArrayを構築し、各区間の右端を取得する方法

Pandasでは、分割点（breaks）の配列からIntervalArray（区間配列）を簡単に構築できます。構築には pandas.arrays.IntervalArray.from_breaks() メソッドを使用し、各区間の右端（右側の端点）を取得するには array.right プロパティを使います。

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

まず、Pandasをインポートします。

import pandas as pd

分割の配列からIntervalArrayを構築する

from_breaks() に分割点のリストを渡すと、隣り合う分割点同士が1つの区間となったIntervalArrayが生成されます。例えば [0, 1, 2, 3, 4, 5] を渡すと、(0, 1]、(1, 2]、(2, 3]、(3, 4]、(4, 5] の5つの区間が作られます。

array = pd.arrays.IntervalArray.from_breaks([0, 1, 2, 3, 4, 5])

作成した区間を表示してみましょう。

print("Our IntervalArray...\n",array)

続いて、right プロパティで各区間の右端を取得します。

print("\nThe right endpoints of each Interval in the IntervalArray as an Index...\n",array.right)

完全なサンプルコード

以下は、区間の長さ（length）や中間点（mid）もあわせて確認できる完全なコード例です。

import pandas as pd

# 分割の配列から新しいIntervalArrayを構築
array = pd.arrays.IntervalArray.from_breaks([0, 1, 2, 3, 4, 5])

# IntervalArrayを表示
print("Our IntervalArray...\n",array)

# IntervalArrayの長さを取得
# 各区間の長さをIndexとして返す
print("\nOur IntervalArray length...\n",array.length)

# 各区間の中間点をIndexとして取得
print("\nThe midpoint of each interval in the IntervalArray...\n",array.mid)

# 各区間の右端をIndexとして取得
print("\nThe right endpoints of each Interval in the IntervalArray as an Index...\n",array.right)

実行結果

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

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

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

The midpoint of each interval in the IntervalArray...
Float64Index([0.5, 1.5, 2.5, 3.5, 4.5], dtype='float64')

The right endpoints of each Interval in the IntervalArray as an Index...
Int64Index([1, 2, 3, 4, 5], dtype='int64')

ポイントのまとめ

• from_breaks()：分割点の配列を受け取り、隣接する分割点ごとに区間を生成してIntervalArrayを構築するメソッドです。
• array.right：各区間の右端の値をInt64IndexなどのIndexオブジェクトとして返します。今回の例では [1, 2, 3, 4, 5] が取得できました。
• array.length：各区間の長さをIndexとして返します。
• array.mid：各区間の中間点を浮動小数点数のIndexとして返します。

デフォルトでは区間は左開き・右閉じ（(a, b] 形式）で生成されるため、右端の値は各区間に含まれる値になります。この挙動は closed 引数で変更可能です。