熊貓新手：如何根據列的內容分配值？

Question

我想創建一個新列，並根據pet1包含單詞“ cat”還是單詞“ dog”分配數值

    pet1           
0   dog             
1   dog;cat;rabbit  
2   cat;dog         
3   manbearpig      
4   hippo           

我希望最終結果如下：

    pet1            points
0   dog             5
1   dog;cat;rabbit  5
2   cat;dog         5
3   manbearpig      0
4   hippo           0 

我該怎么做呢？

我是一個完全編碼的新手，並且一直在嘗試熊貓。 抱歉，這是一個完整的新手問題。

Answer 1

您可以使用contains的字符串方法。
從此數據幀開始：

In [96]: df
Out[96]:
             pet1
0             dog
1  dog;cat;rabbit
2         cat;dog
3      manbearpig
4           hippo

您可以檢查每個元素是否包含子字符串“ dog”：

In [97]: df['pet1'].str.contains('dog')
Out[97]:
0     True
1     True
2     True
3    False
4    False
Name: pet1, dtype: bool

然后乘以5即可得出您想要的結果：

In [98]: df['pet1'].str.contains('dog') * 5
Out[98]:
0    5
1    5
2    5
3    0
4    0
Name: pet1, dtype: int32