python - 查找字符串中的子字符串

Question

我有一个带有“状态”列的数据框。如果状态值包含单词“Won”，我会将状态列的值替换为 1，否则将其替换为 0。

我尝试过这样的事情:

s = oppty_oppline['status']
if s.find("Won") == -1:
       oppty_oppline['status']=0
else:
        oppty_oppline['status']=1

但是我收到了这个错误:

AttributeError                            Traceback (most recent call last)
<ipython-input-23-7d82ff1c0e85> in <module>()
      1 s = oppty_oppline['status']
----> 2 if s.contains("Won") == -1:
      3        oppty_oppline['status']=0
      4 else:
      5         oppty_oppline['status']=1

~/anaconda3/envs/python3/lib/python3.6/site-packages/pandas/core/generic.py in __getattr__(self, name)
   3612             if name in self._info_axis:
   3613                 return self[name]
-> 3614             return object.__getattribute__(self, name)
   3615 
   3616     def __setattr__(self, name, value):

AttributeError: 'Series' object has no attribute 'contains'

知道如何做吗？

Answer 1

使用Series.str.contains对于 bool 掩码，并通过转换为整数将 True/False 映射到 1/0:

oppty_oppline['status'] = oppty_oppline['status'].str.contains('Won').astype(int)

或者使用numpy.where :

oppty_oppline['status'] = np.where(oppty_oppline['status'].str.contains('Won'), 0, 1)