Python Pandas IntervalIndex - Lấy vị trí tất cả khoảng chứa nhãn

Khi một nhãn nằm trong nhiều khoảng chồng chéo, dùng IntervalIndex.get_loc() để lấy vị trí (boolean mask) của tất cả khoảng chứa nhãn đó.

Các bước thực hiện

1. Import thư viện

import pandas as pd

2. Tạo các Interval đóng hai đầu

interval1 = pd.Interval(50, 75, closed='both')
interval2 = pd.Interval(75, 90, closed='both')
interval3 = pd.Interval(50, 90, closed='both')

3. Tạo IntervalIndex

index = pd.IntervalIndex([interval1, interval2, interval3])

4. Lấy vị trí chứa nhãn 65

print("\nVị trí khoảng chứa 65...\n", index.get_loc(65))

Ví dụ hoàn chỉnh

import pandas as pd

interval1 = pd.Interval(50, 75, closed='both')
interval2 = pd.Interval(75, 90, closed='both')
interval3 = pd.Interval(50, 90, closed='both')

print("Interval1...\n", interval1)
print("Interval2...\n", interval2)
print("Interval3...\n", interval3)

index = pd.IntervalIndex([interval1, interval2, interval3])
print("\nVị trí khoảng chứa 65...\n", index.get_loc(65))

Kết quả đầu ra

Interval1...
[50, 75]
Interval2...
[75, 90]
Interval3...
[50, 90]
Vị trí khoảng chứa 65...
[ True False True]

Giải thích: Giá trị 65 nằm trong interval1 [50,75] và interval3 [50,90] → trả về mảng boolean [True, False, True].