python - 根据字符串条件为 Pandas 数据框列赋值

Question

假设我有一个数据框，

data
id  URL
1   www.pandora.com
2   m.jcpenney.com
3   www.youtube.com
4   www.facebook.com

我想根据 URL 是否包含某个特定单词的条件创建一个新列。假设它包含“youtube”，我希望我的列值为 youtube。所以我尝试了以下，

data['test'] = 'other'

所以一旦我们这样做了，

data['test']
other
other
other
other

然后我试了一下，

data[data['URL'].str.contains("youtub") == True]['test'] = 'Youtube'
data[data['URL'].str.contains("face") == True]['test'] = 'Facebook'

虽然这运行没有任何错误，但测试列的值并没有改变。它仍然只对所有列有其他的。当我运行此语句时，理想情况下，第 3 行单独显示更改为“Youtube”，第 4 行显示为“Facebook”。但它没有改变。谁能告诉我我在这里犯了什么错误？

Answer 1

我想你可以使用 loc使用 contains 创建的 bool 掩码:

print data['URL'].str.contains("youtub")
0    False
1    False
2     True
3    False
Name: URL, dtype: bool

data.loc[data['URL'].str.contains("youtub"),'test'] = 'Youtube'
data.loc[data['URL'].str.contains("face"),'test'] = 'Facebook'
print data
   id               URL      test
0   1   www.pandora.com       NaN
1   2    m.jcpenney.com       NaN
2   3   www.youtube.com   Youtube
3   4  www.facebook.com  Facebook