原创 | Python | set_index 和 reset_

import pandas as pd
import numpy as np

df = pd.DataFrame(np.arange(10).reshape(2,5),columns=list('abcde'))

# set_index
df.set_index('a')  # 设置a索引 

    b   c   d   e
a               
0   1   2   3   4
5   6   7   8   9

df.reset_index(['a','b']) # 设置ab两索引

        c   d   e
a   b           
0   1   2   3   4
5   6   7   8   9

df.set_index(['a','b'],drop=False)  # drop默认True

        a   b   c   d   e
a   b                   
0   1   0   1   2   3   4
5   6   5   6   7   8   9

df.set_index(['a','b'],drop=False,inplace=True)  # 覆盖df数据

        a   b   c   d   e
a   b                   
0   1   0   1   2   3   4
5   6   5   6   7   8   9

# reset_index()
df.reset_index()

    a   b   c   d   e
0   0   1   2   3   4
1   5   6   7   8   9

df.reset_index('a')

    a   c   d   e
b               
1   0   2   3   4
6   5   7   8   9

df.reset_index(['a','b']) #设置ab索引列

    a   b   c   d   e
0   0   1   2   3   4
1   5   6   7   8   9

df.reset_index(level=0) # 还原第1个index列

    a   c   d   e
b               
1   0   2   3   4
6   5   7   8   9

df.reset_index(level=1) # 还原第2个index列

    b   c   d   e
a               
0   1   2   3   4
5   6   7   8   9

df.reset_index(level=1,drop=True)  # drop 默认False

    c   d   e
a           
0   2   3   4
5   7   8   9