Python Pandas：DataFrame 列的条件性字符串拆分与拼接技巧-Python教程-PHP中文网

Python Pandas：DataFrame 列的条件性字符串拆分与拼接技巧

心靈之曲

发布： 2025-09-26 14:57:29

原创

929人浏览过

Python Pandas：DataFrame 列的条件性字符串拆分与拼接技巧

本文探讨了在Pandas DataFrame中如何根据列值是否包含特定词语来执行条件性的字符串操作。通过一个实际的地址数据处理场景，详细介绍了如何利用自定义函数结合apply()方法，实现字符串的条件性拆分、清理和重构，避免了不必要的修改，确保数据转换的精确性和灵活性。

1. 引言：DataFrame 列的字符串操作挑战

在数据清洗和预处理中，我们经常需要对dataframe的字符串列进行操作，例如拆分、替换或拼接。然而，当这些操作需要基于特定条件（例如，只有当字符串包含某个子串时才执行）时，直接使用pandas的.str访问器可能无法满足需求，因为它通常会无差别地应用于所有行。本教程将展示如何优雅地解决这一问题。

2. 问题场景：地址列的条件性处理

假设我们有一个包含地址信息的DataFrame，其中address列可能包含“floor”（楼层）信息。我们的目标是：

如果地址中包含“floor”，则将其拆分，保留“floor”之前的部分，然后重新拼接上“ floor”。
如果地址中不包含“floor”，则保持原始地址不变。

原始数据示例：

                  address
0  xxx City yyy road 17 number 8 floor west bank
1                     ttt City iii road 1 number
2     ggg City kkk road 25 number 1 floor apple store

登录后复制

期望结果：

                  address
0  xxx City yyy road 17 number 8 floor
1                     ttt City iii road 1 number
2     ggg City kkk road 25 number 1 floor

登录后复制

3. 常见误区：无条件应用.str方法

许多初学者可能会尝试使用如下代码：

立即学习“Python免费学习笔记（深入）”；

import pandas as pd

data = {
    'address': [
        'xxx City yyy road 17 number 8 floor west bank',
        'ttt City iii road 1 number',
        'ggg City kkk road 25 number 1 floor apple store'
    ]
}
df = pd.DataFrame(data)

# 错误的尝试：无条件拆分和拼接
df['address_attempt'] = df['address'].str.split('floor').str[0] + ' floor'
print("错误的尝试结果：")
print(df[['address', 'address_attempt']])

登录后复制

输出结果：

错误的尝试结果：
                                         address                      address_attempt
0  xxx City yyy road 17 number 8 floor west bank  xxx City yyy road 17 number 8 floor
1                     ttt City iii road 1 number           ttt City iii road 1 number floor
2     ggg City kkk road 25 number 1 floor apple store  ggg City kkk road 25 number 1 floor

登录后复制

分析： 上述代码的问题在于df['address'].str.split('floor').str[0]会应用于所有行。

对于包含“floor”的行，它会正确地获取“floor”之前的部分。
对于不包含“floor”的行（如第二行），split('floor')会返回一个包含原始字符串的列表，例如['ttt City iii road 1 number']。此时str[0]仍然是原始字符串，然后无条件地拼接了' floor'，导致不符合预期的结果。

4. 正确方法：结合自定义函数与apply()

解决此问题的关键是引入条件判断。我们可以定义一个自定义函数，该函数接收单个地址字符串作为输入，并在函数内部进行条件判断，然后使用DataFrame.apply()方法将其应用于DataFrame的整个列。

巧文书

巧文书是一款AI写标书、AI写方案的产品。通过自研的先进AI大模型，精准解析招标文件，智能生成投标内容。

查看详情

示例代码：

import pandas as pd

# 准备数据
data = {
    'address': [
        'xxx City yyy road 17 number 8 floor west bank',
        'ttt City iii road 1 number',
        'ggg City kkk road 25 number 1 floor apple store',
        'aaa City bbb road 10' # 增加一个不含floor的例子
    ]
}
df = pd.DataFrame(data)

# 定义处理地址的函数
def process_address(address):
    """
    根据地址是否包含 'floor' 进行条件性处理。
    如果包含，则拆分、清理并重新拼接 ' floor'；
    否则，返回原始地址。
    """
    if isinstance(address, str) and 'floor' in address:
        # 拆分字符串，取第一部分，去除前后空格，然后拼接 ' floor'
        return address.split('floor')[0].strip() + ' floor'
    else:
        # 如果不包含 'floor' 或不是字符串类型，则返回原始值
        return address

# 应用函数到 'address' 列
df['processed_address'] = df['address'].apply(process_address)

print("最终处理结果：")
print(df[['address', 'processed_address']])

登录后复制

输出结果：

最终处理结果：
                                         address                      processed_address
0  xxx City yyy road 17 number 8 floor west bank  xxx City yyy road 17 number 8 floor
1                     ttt City iii road 1 number           ttt City iii road 1 number
2     ggg City kkk road 25 number 1 floor apple store  ggg City kkk road 25 number 1 floor
3                         aaa City bbb road 10                         aaa City bbb road 10

登录后复制

代码解析：

process_address(address) 函数：
- if isinstance(address, str) and 'floor' in address:：这是一个关键的条件判断。
  - isinstance(address, str)：首先检查输入是否为字符串类型。这对于处理可能存在的非字符串值（如None或NaN）非常重要，可以避免类型错误。
  - 'floor' in address：检查字符串中是否包含子串“floor”。只有当这个条件为真时，才会执行后续的拆分和拼接逻辑。
- address.split('floor')[0]：如果条件满足，将字符串按“floor”拆分，并取第一个元素（即“floor”之前的部分）。
- .strip()：这一步非常重要，用于去除拆分后可能存在的前后空格，确保结果的整洁。
- + ' floor'：将清理后的部分与“ floor”重新拼接起来。注意这里拼接的是' floor'，包含一个空格，以保持格式统一。
- else: return address：如果条件不满足（即不包含“floor”或不是字符串），则直接返回原始的address值，不做任何修改。
df['address'].apply(process_address)：
- apply()是Pandas DataFrame和Series的一个强大方法，它允许你对Series中的每个元素（或DataFrame中的每行/列）应用一个函数。
- 在这里，process_address函数会依次作用于df['address']列的每一个单元格，并将其返回值作为新列processed_address的对应值。

5. 注意事项

strip() 的重要性： 在拆分后使用.strip()可以有效去除多余的空白字符，避免生成如"xxx City yyy road 17 number 8 floor"（多余空格）这样的结果。
大小写敏感性： in 操作符是大小写敏感的。如果需要进行不区分大小写的匹配，可以先将字符串转换为小写或大写（例如address.lower()）。
处理空值（NaN）： apply() 方法在遇到NaN值时，如果自定义函数没有特别处理，可能会导致错误。在process_address函数中增加isinstance(address, str)检查是很好的实践，它能避免对非字符串类型（包括NaN）执行字符串方法。
性能考量： 对于非常大的数据集（数百万行），apply()虽然灵活，但可能不如矢量化操作（如str.contains()结合np.where）高效。然而，对于本例中涉及的复杂条件逻辑，apply()结合自定义函数往往是可读性和性能之间的一个良好平衡点。
多次出现目标词： split('floor')默认是按第一个分隔符进行拆分。如果地址中可能出现多个“floor”且需要不同处理，则需要更复杂的逻辑（例如使用正则表达式或循环）。

6. 总结

通过结合自定义函数和DataFrame.apply()方法，我们可以实现Pandas DataFrame列中复杂且条件性的字符串操作。这种方法提供了高度的灵活性和精确性，确保数据转换能够严格按照业务逻辑进行，避免了不必要的修改，是数据清洗和预处理中非常实用的技巧。在处理字符串数据时，始终考虑潜在的空值、类型不匹配和性能需求，以构建健壮的数据处理流程。

以上就是Python Pandas：DataFrame 列的条件性字符串拆分与拼接技巧的详细内容，更多请关注php中文网其它相关文章！