如何将DataFrame字符串格式索引替换为自定义类的实例

核心原因 #

Pandas的Index对象是不可变序列，设计上不支持单独修改某一位置的索引元素，因此你无法直接给df.index[0]赋值，必须整体替换整个索引。

实现步骤 #

1. 先按原索引的顺序，构造和DataFrame行数长度一致的自定义类实例列表
2. 直接将整个df.index替换为上述实例列表即可，Pandas支持object类型索引存储任意Python对象

代码示例 #

首先模拟你给出的示例DataFrame：

import pandas as pd

# 构造示例df
df = pd.DataFrame({
    'Jane': ['Jane', 25],
    'Max': ['Max', 20]
}, index=['Name', 'Age'])

定义自定义类，可选添加__repr__方法优化后续打印效果：

class I_am_a_class:
    def __init__(self, label):
        self.label = label
    # 自定义打印显示内容，可按需调整
    def __repr__(self):
        return f"I_am_a_class(label={self.label!r})"

替换索引：

# 按顺序构造自定义类实例列表
new_index = [I_am_a_class(label) for label in df.index]
# 整体替换索引
df.index = new_index

验证效果 #

替换完成后，df.index[0]就是I_am_a_class类的实例，可以直接调用类的属性和方法：

print(type(df.index[0]))
# 输出：<class '__main__.I_am_a_class'>
print(df.index[0].label)
# 输出：Name

注意事项 #

• 必须整体替换索引，单独修改df.index[x]会触发报错
• 建议给自定义类添加__repr__方法，否则打印DataFrame时索引会显示对象内存地址，可读性差

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