pandas DataFrame批量生成规则序列索引、避免手动输入的方案咨询

实现代码 #

import pandas as pd

# 替换为你自己的DataFrame对象
# df = pd.DataFrame(index=range(1111))

# 生成前11行索引
first_part = [f"p{i*100:03d}" if i < 10 else "p1000" for i in range(11)]

# 生成中间100行索引
second_part = []
for X in range(1, 11):
    for j in range(1, 11):
        second_part.append(f"p{X}{j*10}")

# 生成最后1000行索引
third_part = []
for block_id in range(1, 11):
    # 块内序号补零为3位，保证格式统一
    for seq in range(1, 101):
        third_part.append(f"p{block_id}{seq:03d}")

# 合并索引并赋值给DataFrame
custom_index = first_part + second_part + third_part
df.index = custom_index

# 校验长度，输出1111即为匹配成功
print(len(custom_index))

规则匹配说明 #

• 前11行完全覆盖p000到p1000的要求格式
• 中间100行每个X取值1-10，对应生成pX10到pX100格式的索引
• 最后1000行分为10个百行块，p后第一位依次为1到10，和要求的块编号替换规则完全匹配

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