如何从DataFrame中部分移除索引值及转换pivot_table生成的DataFrame

嘿，这两个问题我实际工作中经常碰到，给你整理了具体的解决方法👇

根据你要移除的索引类型（标签/位置），可以用这几种实用方法：

• 按索引标签移除
  直接用drop()方法指定要移除的索引标签，默认返回新的DataFrame（如果要直接修改原数据，加上inplace=True参数即可）：

  import pandas as pd
  
  # 示例DataFrame
  df = pd.DataFrame({'value': [1,2,3,4]}, index=['a','b','c','d'])
  # 移除标签为'b'和'd'的索引
  new_df = df.drop(index=['b','d'])
  

• 按索引位置移除
  如果不知道索引标签，只知道要移除的位置，可以通过df.index获取索引对象后指定位置：

  # 移除第1和第3个位置的索引（索引从0开始计数）
  new_df = df.drop(df.index[[1, 3]])
  

• 处理多级索引（MultiIndex）
  要是遇到多级索引的情况，可以通过level参数指定要操作的索引层级：

  # 示例多级索引DataFrame
  multi_df = pd.DataFrame({'value': [1,2,3,4]}, 
                          index=pd.MultiIndex.from_tuples([('x','a'), ('x','b'), ('y','c'), ('y','d')],
                                                           names=['level1', 'level2']))
  # 移除level1中标签为'x'的所有索引
  new_multi_df = multi_df.drop(index='x', level='level1')
  

首先得明确“指定格式”的具体需求，我举两个最常见的场景，你可以按需调整：

场景1：从宽格式转长格式（最常用） #

假设你的pivot_table生成的是类似这样的宽表：

import pandas as pd

# 原始数据
data = {'Name': ['Alice', 'Bob', 'Alice', 'Bob'],
        'Subject': ['Math', 'Math', 'English', 'English'],
        'Score': [85, 90, 78, 82]}
# 生成pivot_table
pivot_df = pd.pivot_table(data, index='Name', columns='Subject', values='Score')
# pivot_df的结构：
# Subject  English  Math
# Name
# Alice        78    85
# Bob          82    90

要转成Name、Subject、Score的长格式，用stack()或者melt()都可以：

# 方法1：stack()，把列索引转成行索引后重置索引
long_df = pivot_df.stack().reset_index(name='Score')

# 方法2：melt()，更直观指定保留的核心列
long_df = pivot_df.reset_index().melt(id_vars='Name', var_name='Subject', value_name='Score')

转换后的结果：

Name  Subject  Score
0  Alice  English     78
1    Bob  English     82
2  Alice     Math     85
3    Bob     Math     90

场景2：自定义宽格式（调整索引与列） #

如果只是要把行索引转成普通列，或者调整列的顺序，用reset_index()配合列重命名/排序就行：

# 把行索引Name转成普通列
new_df = pivot_df.reset_index()
# 调整列的显示顺序
new_df = new_df[['Name', 'Math', 'English']]

如果你的“指定格式”是其他特殊结构，比如拆分多级索引、合并列等，可以结合reset_index()、rename()、concat()这些方法组合使用——核心思路是先把索引转成普通列，再按需调整结构。

内容的提问来源于stack exchange，提问作者Rathi Rao