如何在python / pandas中将一个字符串从一个df列添加到另一个df列？

Question

我正在使用一个pandas数据框，该数据框包含10.000多个指向报纸文章的链接。我想在我的链接中使用字符串的一部分来显示文章的主题/编辑，例如/ media /或/ sport /，到另一列。

实施例： Col 1包含以下链接： www.theguardian.com/us-news/wikileaks-during-final-stages-of-election www.theguardian.com/politics/parliament-to-have-final -

即。我想制作一个def来迭代Col 1中的字符串，如果一个字符串包含“/ us-news /”，那么“US News”就会添加Col 2，就好像一个字符串包含/ politics / then“Politics”是添加了Ti Col 2.

有谁知道怎么做？

Answer 1

numpy.select使用str.contains条件：

m1 = df['col1'].str.contains('/us-news/')
m2 = df['col1'].str.contains('/politics/')

df['col2'] = np.select([m1,m2], ['US News','Politics'], default='no match')