pandas 给dataframe添加列名的两种方法

以下是详细的实例教程，包含两种添加列名的方法和示例说明。

方法一：在生成 dataframe 时指定列名

生成 dataframe 时可以在构造器中指定列名，例如：

import pandas as pd
import numpy as np

data = [
    [1, 'A', 10],
    [2, 'B', 20],
    [3, 'C', 30],
    [4, 'D', 40]
]

columns = ['id', 'name', 'age']

df = pd.DataFrame(data=data, columns=columns)
print(df)

上述代码中，我们通过指定 columns 参数来设置 df 中的列名。运行结果如下：

   id name  age
0   1    A   10
1   2    B   20
2   3    C   30
3   4    D   40

方法二：使用 rename() 方法为 dataframe 添加列名

另一种添加列名的方式是使用 pandas 中的 rename() 方法。在使用此方法时，我们需要为 dataframe 的每一列分别指定列名，然后将这些列名传递给 rename() 方法。

以下是示例代码：

import pandas as pd
import numpy as np

data = [
    [1, 'A', 10],
    [2, 'B', 20],
    [3, 'C', 30],
    [4, 'D', 40]
]

df = pd.DataFrame(data=data)
print(df)

# 设置列名
columns = ['id', 'name', 'age']

df = df.rename(columns={i: c for i, c in enumerate(columns)})
print(df)

运行结果如下：

   0  1   2
0  1  A  10
1  2  B  20
2  3  C  30
3  4  D  40

   id name  age
0   1    A   10
1   2    B   20
2   3    C   30
3   4    D   40

在上述代码中，我们生成了一个没有列名的 dataframe，然后使用 rename() 方法为其设置了列名。具体来说，我们先定义了一个 columns 变量，其中包含了 dataframe 的列名信息。然后，我们通过 rename() 方法与字典推导式将原 dataframe 的列名与新的列名对应起来，并将其赋值给原 dataframe。最终得到了带有列名的 dataframe。

以上就是加入列名的两种方法的示例说明。