Pandas DataFrame 索引修改：使用 set_index() 方法

Pandas 中的 DataFrame 可以使用 set_index() 方法来改变索引。set_index() 方法接受一个或多个列名或列的索引，将其作为新的索引。

例如，假设我们有以下 DataFrame：

import pandas as pd

df = pd.DataFrame({
    'name': ['Alice', 'Bob', 'Charlie', 'David'],
    'age': [25, 30, 35, 40],
    'salary': [50000, 60000, 70000, 80000]
})

print(df)

输出：

       name  age  salary
0     Alice   25   50000
1       Bob   30   60000
2   Charlie   35   70000
3     David   40   80000

我们可以使用 set_index() 方法将 'name' 列作为新的索引：

df = df.set_index('name')

print(df)

输出：

         age  salary
name               
Alice     25   50000
Bob       30   60000
Charlie   35   70000
David     40   80000

我们还可以使用多个列来创建新的索引。例如，如果我们想使用 'name' 和 'age' 列作为新的索引：

df = df.set_index(['name', 'age'])

print(df)

输出：

            salary
name    age       
Alice   25    50000
Bob     30    60000
Charlie 35    70000
David   40    80000