首页 > 解决方案 > 返回包含字符串的列的布尔掩码

问题描述

我有一个带有 5 列的 Pandas 数据框。一些列包含字符串“aa”。我想返回一个包含布尔值的列表,这些布尔值标识包含“XX”的列示例

     cl1  cl2  cl3  cl4  cl5  
1.   ds   fr   we   iu   XX     
2.   sx   XX   ee   ee   nn
3.   df   vd   rr   we   mm
4.   er   we   gf   wq   hh

结果应该是

res= [False True False False True]

我尝试使用 str.match 的 str.contain 但这仅适用于系列,我不想遍历系列

标签: pythonpandasselectmatch

解决方案


你可以这样做:

res = (df == 'XX').any(0).tolist()

推荐阅读