Computer >> บทช่วยสอนคอมพิวเตอร์ >  >> การเขียนโปรแกรม >> Python

Python Pandas - คืน MultiIndex ที่ลบหลายระดับโดยใช้ชื่อระดับ

บทนำ

ใช้ MultiIndex.droplevel() พร้อมส่ง รายการชื่อระดับ เพื่อลบหลายระดับพร้อมกัน

ขั้นตอน

1. นำเข้าไลบรารีและสร้าง MultiIndex 3 ระดับ

import pandas as pd
arrays = [[2, 4, 3, 1], ['Peter', 'Chris', 'Andy', 'Jacob'], [50, 30, 40, 70]]
multiIndex = pd.MultiIndex.from_arrays(arrays, names=('rank', 'student', 'points'))

2. ลบระดับ 'rank' และ 'student' พร้อมกัน

print("\nMulti-index after dropping two levels...\n",
      multiIndex.droplevel(['rank', 'student']))

ตัวอย่างโค้ดเต็ม

import pandas as pd

arrays = [[2, 4, 3, 1], ['Peter', 'Chris', 'Andy', 'Jacob'], [50, 30, 40, 70]]
multiIndex = pd.MultiIndex.from_arrays(arrays, names=('rank', 'student', 'points'))

print("The Multi-index...\n", multiIndex)
print("\nThe levels in Multi-index...\n", multiIndex.levels)
print("\nMulti-index after dropping two levels...\n",
      multiIndex.droplevel(['rank', 'student']))

ผลลัพธ์

The Multi-index...
MultiIndex([(2, 'Peter', 50), (4, 'Chris', 30), (3, 'Andy', 40), (1, 'Jacob', 70)],
           names=['rank', 'student', 'points'])

The levels in Multi-index...
[[1, 2, 3, 4], ['Andy', 'Chris', 'Jacob', 'Peter'], [30, 40, 50, 70]]

Multi-index after dropping two levels...
Int64Index([50, 30, 40, 70], dtype='int64', name='points')

คำอธิบาย

  • droplevel(['rank', 'student']) ลบระดับที่ระบุ ทั้งสองระดับพร้อมกัน
  • เหลือระดับ 'points' เท่านั้น → กลายเป็น Int64Index ธรรมดา
  • รองรับการส่ง list ของชื่อระดับ หรือ list ของ index ตำแหน่ง