Python Pandas - Lấy số nguyên cấp độ trong MultiIndex

Để lấy số nguyên đại diện cho số lượng cấp độ trong MultiIndex, chúng ta sử dụng thuộc tính MultiIndex.nlevels.

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

1. Import thư viện

import pandas as pd

2. Tạo MultiIndex

arrays = [[1, 2, 3, 4, 5], ['John', 'Tim', 'Jacob', 'Chris', 'Keiron']]
multiIndex = pd.MultiIndex.from_arrays(arrays, names=('ranks', 'student'))

3. Lấy số cấp độ

print("\nSố cấp độ trong MultiIndex:\n", multiIndex.nlevels)

Ví dụ hoàn chỉnh

import pandas as pd

arrays = [[1, 2, 3, 4, 5], ['John', 'Tim', 'Jacob', 'Chris', 'Keiron']]
multiIndex = pd.MultiIndex.from_arrays(arrays, names=('ranks', 'student'))

print("MultiIndex:\n", multiIndex)
print("\nSố cấp độ (nlevels):\n", multiIndex.nlevels)
print("\nCác cấp độ (levels):\n", multiIndex.levels)

Kết quả đầu ra

MultiIndex:
MultiIndex([(1, 'John'), (2, 'Tim'), (3, 'Jacob'), (4, 'Chris'), (5, 'Keiron')], names=['ranks', 'student'])
Số cấp độ (nlevels):
2
Các cấp độ (levels):
[[1, 2, 3, 4, 5], ['Chris', 'Jacob', 'John', 'Keiron', 'Tim']]

Ứng dụng thực tế

# Kiểm tra số cấp độ trước khi xử lý
if multiIndex.nlevels == 2:
    # Xử lý MultiIndex 2 cấp
    pass
elif multiIndex.nlevels > 2:
    # Xử lý MultiIndex nhiều cấp
    pass

# Lặp qua các cấp độ
for i in range(multiIndex.nlevels):
    print(f"Cấp độ {i}: {multiIndex.levels[i]}")

Thuộc tính nlevels rất hữu ích khi viết code generic xử lý MultiIndex với số cấp độ không cố định.