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

  • Post category:Python

下面是详细的讲解,包含两条添加列名的实例:

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

在 pandas 中,添加列名可以通过两种方法实现:

  1. 直接更改 columns 属性
  2. 使用 rename()函数

方法一:直接更改 columns 属性

下面我们来看一个实例,首先创造一个 3 行 4 列的 dataframe:

import pandas as pd
import numpy as np

data = np.random.randn(3, 4)
df = pd.DataFrame(data, columns=['A','B','C','D'])
print(df)

输出结果如下:

          A         B         C         D
0 -0.022720  0.092614 -0.939206 -0.393872
1  2.623221 -0.611706 -1.096951  0.855114
2  1.317496  1.259154 -0.102474  0.475360

接下来,我们更改列名:

df.columns=['aa', 'bb', 'cc', 'dd']
print(df)

输出结果为:

         aa        bb        cc        dd
0 -0.022720  0.092614 -0.939206 -0.393872
1  2.623221 -0.611706 -1.096951  0.855114
2  1.317496  1.259154 -0.102474  0.475360

我们可以看到,通过更改 columns 属性,我们成功地给 dataframe 添加了新的列名。

方法二:使用 rename()函数

除了直接更改 columns 属性,我们还可以使用 rename() 函数。下面我们来看一个实例:

df = pd.DataFrame(data, columns=['A','B','C','D'])
df = df.rename(columns={'A':'aa','B':'bb'})
print(df)

输出结果为:

         aa        bb         C         D
0 -0.022720  0.092614 -0.939206 -0.393872
1  2.623221 -0.611706 -1.096951  0.855114
2  1.317496  1.259154 -0.102474  0.475360

我们可以看到,通过 rename() 函数,我们同样成功地给 dataframe 添加了新的列名。

至此,完成了 pandas 给 dataframe 添加列名的两种方法的详细实例教程。