如何基于DataFrame的utm_Campaign列修改Channel列的值？

嘿，这个需求我经常遇到，给你分享几个实用的实现方案，你可以根据自己的场景挑最合适的：

这种方法最适合多条件分支的场景，规则一目了然，而且性能比apply好很多，尤其是数据量大的时候。

步骤很简单：先定义条件列表和对应的结果列表，再用select批量赋值，还能指定默认值（比如保留原Channel的值）。

代码示例：

import numpy as np
import pandas as pd

# 模拟你的DataFrame
df = pd.DataFrame({
    'utm_Campaign': ['FacebookPaid', 'Newsletter', 'YoutubePaid', 'GoogleSearch', 'FacebookFree'],
    'Channel': ['', '', '', '', '']
})

# 定义匹配条件和对应输出
conditions = [
    # 匹配所有Facebook相关的关键词，忽略大小写
    df['utm_Campaign'].str.contains('Facebook|FacebookPaid|FacebookFree', case=False),
    df['utm_Campaign'].str.contains('Newsletter', case=False),
    df['utm_Campaign'].str.contains('YoutubePaid', case=False)
]
choices = ['FB', 'Email', 'Youtube Direct']

# 给Channel列赋值，不满足条件的保留原内容
df['Channel'] = np.select(conditions, choices, default=df['Channel'])

后续如果要加新规则，直接在conditions和choices里加一行就行，维护起来特别方便。

如果你的规则后续可能有更复杂的逻辑（比如要结合其他列判断），apply会更灵活，但数据量很大的时候，它的性能不如numpy.select。

代码示例：

def assign_channel(row):
    campaign = row['utm_Campaign']
    # 按规则判断
    if any(keyword in campaign for keyword in ['Facebook', 'FacebookPaid', 'FacebookFree']):
        return 'FB'
    elif 'Newsletter' in campaign:
        return 'Email'
    elif 'YoutubePaid' in campaign:
        return 'Youtube Direct'
    else:
        # 不满足条件就保留原Channel值
        return row['Channel']

# 按行应用函数
df['Channel'] = df.apply(assign_channel, axis=1)

这个写法更贴近常规的逻辑判断，新手更容易理解，小数据量用着完全没问题。

如果你的规则特别简单，也可以用replace配合正则来实现，但多规则的情况下可读性不如前两种方法。

代码示例：

# 用正则匹配包含指定关键词的内容，替换成对应Channel值
df['Channel'] = df['utm_Campaign'].str.replace(
    r'.*(Facebook|FacebookPaid|FacebookFree).*', 'FB', regex=True
).str.replace(
    r'.*Newsletter.*', 'Email', regex=True
).str.replace(
    r'.*YoutubePaid.*', 'Youtube Direct', regex=True
)

# 处理不匹配的情况：如果结果不在目标值里，保留原Channel值
df['Channel'] = df['Channel'].where(
    df['Channel'].isin(['FB', 'Email', 'Youtube Direct']), 
    df['Channel']
)

这个方法适合快速实现简单规则，但正则写起来容易出错，后续修改规则也不够直观。

总结一下 #

• 追求清晰性和性能：选numpy.select()
• 需要复杂自定义逻辑：选df.apply()
• 规则简单且快速实现：可以试试replace的正则写法

内容的提问来源于stack exchange，提问作者mattylim