Computer >> Hướng Dẫn Máy Tính >  >> Lập Trình >> Python

Python Pandas IntervalIndex - Lấy vị trí nguyên cho nhãn


Dùng IntervalIndex.get_loc() để lấy vị trí nguyên (integer location) của khoảng chứa nhãn. Nếu khoảng không chồng chéo, trả về số nguyên duy nhất.

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

1. Import thư viện

import pandas as pd

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

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

3. Tạo IntervalIndex

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

4. Lấy vị trí nguyên cho nhãn 75

print("\nVị trí nguyên cho nhãn 75...\n", index.get_loc(75))

Ví dụ hoàn chỉnh

import pandas as pd

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

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

index = pd.IntervalIndex([interval1, interval2])
print("\nVị trí nguyên cho nhãn 75...\n", index.get_loc(75))

Kết quả đầu ra

Interval1...
[50, 75]
Interval2...
[75, 90]
Vị trí nguyên cho nhãn 75...
0

Giải thích: Với closed='both', 75 thuộc cả hai khoảng. get_loc trả về vị trí đầu tiên khớp (index 0). Nếu cần tất cả, dùng boolean mask như bài trước.