python - 如何根据另一列更改数据框列中的值？

Question

我有一个数据框，我想将函数传递到特定列:

希望根据某些要求更改“ channel ”列值:

if 'Facebook' or 'FacebookPaid' or 'FacebookFree' in df['utm_Campaign']:
    df['Channel'] = 'FB'
elif 'Newsletter' in df['utm_Campaign']:
    df['Channel'] = 'Email'
else:
   if 'YoutubePaid' in df['utm_Campaign']:
                df['Channel'] = 'Youtube Direct'

基本上，如果同一行的另一列 (utm_Campaign) 中包含单词“Facebook”，则数据帧的“ channel ”列中的字符串将设置为“FB”。

不知道该怎么办。我是否映射一个函数，或者执行 df.replace，或者其他什么？

Answer 1

您可以通过 str.contains 创建条件然后使用 numpy.select :

m1 = df['utm_Campaign'].str.contains('Facebook|FacebookPaid|FacebookFree')
m2 = df['utm_Campaign'].str.contains('Newsletter')
m3 = df['utm_Campaign'].str.contains('YoutubePaid')

df['Channel'] = np.select([m1,m2,m3], ['FB','Email','Youtube Direct'], default='no match')