【随笔】使用pandas.reset_index() 一些坑

前言

使用dataframe.reset_index()，一些坑，主要针对reset之后index变成column的列名问题

代码

比如我直接reset_index()

列名就是默认index

如果我们用一列来作为index（本质是将一个series当作index，同理），之后再reset_index，就不是默认的了

还是保持全本的列名

此处的series可以自己尝试一下，比如一个dataframe中取出一列就是series，把这一列当作index，看结果

import pandas as pd
data = pd.DataFrame({
    
    'A':[1,2,3,4], 'B':[2,3,4,5], 'woniu':[11,22,33,44]})
print(data.head())
print(data.reset_index())
data = data.set_index('woniu')
print(data.head())
print(data.reset_index())