python - 根据谓词将值添加到新列

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

我将如何浏览数据框、过滤公司列并将值提取到新列中？像这样的事情:

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

Answer 1

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

#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')