根据谓词将值添加到新列

Question

假设我有一个由一群人和公司组成的数据框，他们隶属于：

Id|Name  |Company        
1 |Hanzo |Bank of America
2 |Bertha|Disney         
3 |Bob   |Swiss Bank   

现在，我想添加一个新的公司类型的列，它们属于：

Id|Name  |Company        |Type
1 |Hanzo |Bank of America|Bank
2 |Bertha|Disney         |Entertainment
3 |Bob   |Swiss Bank     |Bank

我将如何浏览数据框，过滤company列并将值提取到新列中？ 像这样：

for i in range(len(dataset)):
    if dataset[['Company'].str.contains('BANK')][i]:
        dataset['Type'] = 'Bank'
    else:
        dataset['Type'] = 'Unknown'

Answer 1

如果需要处理2个或更多条件，请使用numpy.select ：

#mask for Entertainment type
ent_companies = ['Disney', ...]
mask1 = dataset['Company'].isin(ent_companies)
#mask for Bank type
mask2 = dataset['Company'].str.contains('BANK', case=False)
dataset['type'] = np.select([mask1, mask2], ['Entertainment', 'Bank'],default='Unknown')

对于进程来说，只有一种更好的条件是numpy.where ：

dataset['type'] = np.where(mask1, 'Bank', 'No bank')