我有一个数据框(系列)
'df'
如下所示:
Name
A1001
A1002
B1001
C1001
A1003
B1002
B1003
C1002
D1001
D1002
我想创建一个名为Company的新专栏
'Alpha'
如果文本以
'A'
,
'Bravo'
如果文本以
'B'
和
'Others'
如果文本以其他内容开头。
我试过:
df['Company'] = 'Alpha' if df.Name.str.startswith('A') else 'Other'
但它给了我一个错误:
ValueError: The truth value of a Series is ambiguous. Use a.empty, a.bool(), a.item(), a.any() or a.all().
预期产量:
Name Company
A1001 Alpha
A1002 Alpha
B1001 Bravo
C1001 Other
A1003 Alpha
B1002 Bravo
B1003 Bravo
C1002 Other
D1001 Other
D1002 Other
怎么能做到?