给定一个dataframe和一个dict如下:
Col A Col B
f3s34Roger nan
bll23Dawn nan
Misty3r2w nan
efJunea4l nan
vendor_dict = {'Roger': 'cups','Charlie':'plates','Misty':'bowls'}
我想搜索每一行,看看Col 'A'是否包含dict中的键。如果包含,我想将Col 'B'设置为键的值。
只使用静态值进行测试,这是可行的:
df['Col B'] = np.where(df['Col A'].str.find('Roger'),'', 'cups')
但这并没有给予我dictMap。我试过:
df['Col B'] = np.where(df['Col A'].str.find().map(vendors)
但这是不对的有什么想法吗
1条答案
按热度按时间aelbi1ox1#
使用
str.findall
然后使用map