Pandas 替换列名中的子字符串 - Python

在处理数据时，经常需要对列名进行修改。在 pandas 中，我们可以使用 rename()方法来修改列名。本文将着重介绍如何替换列名中的子字符串。

代码示例

我们先看一个简单的示例：

import pandas as pd

data = {'Name': ['John', 'Tim', 'Alice'],
        'Age': [25, 30, 35],
        'Gender': ['Male', 'Male', 'Female']}

df = pd.DataFrame(data)

print(df)

# 使用 rename() 方法来替换列名中的子字符串
df = df.rename(columns={'Name': 'First Name'})  # 将 'Name' 列改为 'First Name'

print(df)

输出：

    Name  Age  Gender
0   John   25    Male
1    Tim   30    Male
2  Alice   35  Female

  First Name  Age  Gender
0       John   25    Male
1        Tim   30    Male
2      Alice   35  Female

可以看到，我们使用 rename() 方法来修改了列名，将 'Name' 列改为 'First Name' 列。

接下来，我们来看如何替换列名中的子字符串。

import pandas as pd

data = {'Name': ['John', 'Tim', 'Alice'],
        'Age': [25, 30, 35],
        'Gender': ['Male', 'Male', 'Female']}

df = pd.DataFrame(data)

print(df)

# 将列名中的 'e' 替换为 'E'
df.columns = df.columns.str.replace('e', 'E')

print(df)

输出：

    Name  Age  Gender
0   John   25    Male
1    Tim   30    Male
2  Alice   35  Female

    NamE  AgE  GendEr
0   John   25    Male
1    Tim   30    Male
2  AlicE   35  Female

可以看到，我们使用了 str.replace() 方法来替换列名中的子字符串。

总结

使用 pandas，我们可以方便地进行数据处理。本文介绍了如何替换列名中的子字符串，希望能对大家有所帮助。