如何自定义DataFrame索引：命名为PokeID并实现带字符串的自增序列

嘿，我来帮你解决这个自定义索引的需求！要实现命名为"PokeID"、格式是「字符串+从1开始的自增数字」的索引，其实很简单，分几步就能完成：

1. 生成符合格式的自增索引序列 #

首先，我们需要根据DataFrame的行数，生成对应的索引值。比如你想要前缀是Poke，后面跟三位数字（像Poke001这种），可以用列表推导式快速生成：

# 假设你的DataFrame叫df
custom_index = [f"Poke{i:03d}" for i in range(1, len(df)+1)]

如果不需要固定位数，直接写成f"Poke{i}"就行，这样会得到Poke1、Poke2这种简洁格式。

2. 设置索引并命名 #

把刚才生成的序列设为DataFrame的索引，同时给索引列命名为"PokeID"：

df.index = custom_index
df.index.name = "PokeID"

完整示例演示 #

给你个具体的例子，方便你对照：

import pandas as pd

# 模拟你的原始数据
data = {"精灵名称": ["妙蛙种子", "妙蛙草", "妙蛙花"], "属性": ["草/毒", "草/毒", "草/毒"]}
df = pd.DataFrame(data)

# 生成自定义索引
custom_index = [f"Poke{i:03d}" for i in range(1, len(df)+1)]
# 设置索引和名称
df.index = custom_index
df.index.name = "PokeID"

print(df)

运行后你会得到这样的输出：

精灵名称    属性
PokeID             
Poke001  妙蛙种子  草/毒
Poke002  妙蛙草   草/毒
Poke003  妙蛙花   草/毒

更简洁的写法 #

其实可以一步到位，用pd.Index直接创建带命名的索引：

df.index = pd.Index([f"Poke{i:03d}" for i in range(1, len(df)+1)], name="PokeID")

后续新增行保持自增的小技巧 #

如果之后要给DataFrame加新行，想让索引继续按格式自增，可以这么做：

# 新增一行数据
new_data = {"精灵名称": "小火龙", "属性": "火"}
# 提取当前最后一个索引的数字部分，加1后生成新索引
last_num = int(df.index[-1].replace("Poke", ""))
new_index = f"Poke{last_num + 1:03d}"
# 添加新行
df.loc[new_index] = new_data

这样新增的行索引就会是Poke004，完美延续之前的格式~

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