pandas 为列名添加前缀 - Python

在使用 Pandas 进行数据分析的过程中，经常需要对 DataFrame 中的特定列添加前缀，以方便后续数据处理操作。Pandas 提供了丰富的功能，使得在数据处理时可以轻松地添加前缀。

使用 add_prefix() 函数添加前缀

Pandas 中的 add_prefix() 函数可以很轻松地为 DataFrame 的每一列添加前缀。可以通过如下代码完成：

import pandas as pd

# 构建 DataFrame
df = pd.DataFrame({
    'A': [1, 2, 3],
    'B': [4, 5, 6]
})

# 使用 add_prefix() 函数为 DataFrame 中每一列添加前缀
df = df.add_prefix('prefix_')

# 输出 DataFrame
print(df)

输出结果如下：

   prefix_A  prefix_B
0         1         4
1         2         5
2         3         6

可以看出，此时 DataFrame 中的每一列都已经添加了前缀。

使用 rename() 函数添加前缀

Pandas 还提供了 rename() 函数，可以更加灵活地为 DataFrame 的列添加前缀。可以通过这样的代码实现：

import pandas as pd

# 构建 DataFrame
df = pd.DataFrame({
    'A': [1, 2, 3],
    'B': [4, 5, 6]
})

# 使用 rename() 函数为 DataFrame 中每一列添加前缀
df = df.rename(columns=lambda x: 'prefix_' + x)

# 输出 DataFrame
print(df)

输出结果与之前相同：

   prefix_A  prefix_B
0         1         4
1         2         5
2         3         6

可以看出，使用 rename() 函数也能够为 DataFrame 的每一列添加前缀。

总结

在 Pandas 中，为 DataFrame 的列添加前缀是非常常见的数据处理操作。使用 add_prefix() 和 rename() 函数可以方便地实现此类操作，使得数据处理更加灵活、高效。