reset_index()函数
作用:使用索引重置生成一个新的DataFrame或Series,可以把索引用作列。
话不多说直接上代码
>>> import pandas as pd
>>> import numpy as np
>>> df = pd.DataFrame(data={'A':[1,2,3],'B':[4,5,6],'C':[7,8,9]})
>>> df
A B C
0 1 4 7
1 2 5 8
2 3 6 9
>>> df.reset_index() # 会将原来的索引index作为新的一列
index A B C
0 0 1 4 7
1 1 2 5 8
2 2 3 6 9
>>> df
A B C
0 1 4 7
1 2 5 8
2 3 6 9
>>> df.reset_index(drop=True) # 使用drop参数设置去掉原索引
A B C
0 1 4 7
1 2 5 8
2 3 6 9
总结
在机器学习中常常会对索引进行一定的处理,是否保留原有的索引,可以参照上面的做法,简单易懂。
欢迎交流~