列表元素与 pandas 列的关键字匹配

我的元素列表如下：

 A=  ['loans','s-class','veyron','trump','rihana','drake','election']

我还有另一个 pandas 数据框B有柱子category and words这是逗号分隔的字符串：-

category              words
audi                  a4, a6
bugatti               veyron, chiron
mercedez              s-class, e-class
dslr                  canon, nikon
apple                 iphone,macbook,ipod
finance               sales,loans,sales price
politics              trump, election, votes
entertainment         spiderman,thor, ironmen
music                 beiber, rihana,drake
........              ..............
.........             .........

我想要映射列表的所有元素A带柱words并分配对应的category到一个新列表中。所以，预期输出是。

matched_categories=['finance','mercedez','bugatti','politics','music','music','politics']

过滤boolean indexing http://pandas.pydata.org/pandas-docs/stable/indexing.html#boolean-indexing with iat http://pandas.pydata.org/pandas-docs/stable/generated/pandas.Series.iat.html选择第一个匹配的值：

#if always matched all values
matched_categories = [df.loc[df['words'].str.contains(x), 'category'].iat[0] for x in A]
print (matched_categories)
['finance', 'mercedez', 'bugatti', 'politics', 'music', 'music', 'politics']

如果某个值不匹配，则更通用的解决方案 - 然后返回not matched value:

#added last aaa value
A = ['loans','s-class','veyron','trump','rihana','drake','election','aaa']

matched_categories = [next(iter(df.loc[df['words'].str.contains(x),'category']),'not matched')
                      for x in A]
print (matched_categories)
['finance', 'mercedez', 'bugatti', 'politics', 'music', 'music', 'politics', 'not matched']