python - 检查 pandas 数据框中是否存在确切的字符串

Question

我在 Pandas 中有以下数据框

 code        skills  
 12          sql, r, hadoop
 13          python
 14          r, matlab
 15          ruby, python

我想标记 1 和 0，其中我发现 r 的确切出现如下是我想要的数据帧。

 code        skills             r  
 12          sql, r, hadoop     1   
 13          python             0
 14          r, matlab          1
 15          ruby, python       0

我正在用 pandas 跟随

 df['r'] = df['skills'].eq(' r,').astype(int)

Answer 1

您可以使用一些regex 来完成这个任务。

pattern = '(?:, r)|(?:r,)'

df['r'] = df.skills.str.contains(pattern).astype(int)

   code          skills  r
0    12  sql, r, hadoop  1
1    13          python  0
2    14       r, matlab  1
3    15    ruby, python  0