如何转换Pandas数据框架中某一列的索引

  • Post category:Python

首先需要确保使用的是Pandas版本>=0.21。可以使用以下代码检查版本:

import pandas as pd
print(pd.__version__)

如果版本低于0.21,请先升级到0.21或以上版本。

下面以一个示例数据为例,演示如何转换Pandas数据框架中某一列的索引。

import pandas as pd

df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6], 'C': [7, 8, 9]})
print(df)

示例数据输出如下:

   A  B  C
0  1  4  7
1  2  5  8
2  3  6  9

假设现在需要把列”B”设置为索引列。可以使用以下代码:

df = df.set_index('B')
print(df)

输出结果如下:

   A  C
B      
4  1  7
5  2  8
6  3  9

可以发现,列”B”已经被设置为索引列。

如果需要把索引列恢复为普通列,可以使用以下代码:

df = df.reset_index()
print(df)

输出结果如下:

   B  A  C
0  4  1  7
1  5  2  8
2  6  3  9

可以发现,索引列”B”已经恢复为了普通列。