python - 删除几列后从 0 开始重新分配列标签

Question

我通过以下命令删除了一些重复的列。

columns = XY.columns[:-1].tolist()
XY1 = XY.drop_duplicates(subset=columns,keep='first').

结果如下:

Combined Series shape :  (100, 4)                            
Combined Series:     1    222  223          0                
0    0    0    0  1998.850000                                
1    0    0    0     0.947361
2    0    0    0     0.947361
3    0    0    0     0.947361
4    0    0    0     0.947361

Now the columns is labelled 1 222 223 0 (0 label at the end is because of concat with another df !!) I want the columns to be re-labelled from index 0 onwards. How'll I do it?

Answer 1

因此，首先使用您想要的映射创建一个字典

trafo_dict = {x:y for x,y in zip( [1,222,223,0],np.linspace(0,3,4))}

然后您需要重命名列。这可以通过 pd.DataFrame.rename 来完成:

  XY1 = XY1.rename(columns=trafo_dict)

编辑:如果您希望它以更通用的方式使用:

np.linspace(0,XY1.shape[1]-1,XY1.shape[1])