Pandas数据帧搜索字符串并返回False值

Question

我有一个类似的数据帧

Index    A
0      1  cat
1      2  dog
2      3  bot
3      4  fly

我想创建两列，这取决于列A是否包含字母"A或b或c">

预期结果：

Index| A | yes |   no    |
--------------------------
1    |cat| cat |         |
2    |dog|     |   dog   | 
3    |bot| bot |         |
4    |fly|     |   fly   |

目前我有

abc = ['a', 'b', 'c']
abc = '|'.join(abc)
df['yes'] = df[df['A'].str.contains(abc)]['A']
df['no'] = df[df['A'].str.contains(abc) == False]['A']

选择"是"时效果良好，但选择"否"时，会出现以下错误

ValueError: too many values to unpack (expected 3)

列已创建，但由于出现错误，未来的函数(例如info(((似乎因此而中断。

这可能是因为对a、b和c的检查给出了3个结果。在这种情况下，有没有方法正确返回False值？感谢

Answer 1

这看起来像是set_index和unstack:的作业

m = df['A'].str.contains(abc).replace({True: 'yes', False: 'no'})
m

0    yes
1     no
2    yes
3     no
Name: A, dtype: object
df.set_index(['Index', m])['A'].unstack(fill_value='')
A       no  yes
Index          
1           cat
2      dog     
3           bot
4      fly     

Answer 2

使用您的方法，对代码进行微小的更改，您可以获得正确的输出。

abc = ['a', 'b', 'c']
abc = '|'.join(abc)
df['yes'] = df[df['A'].str.contains(abc)]['A']
df['no'] = df[~df['A'].str.contains(abc)]['A']

Answer 3

另一种方式；

链np.where、str.contains和''.join()以填充"是"one_answers"否"。pivot帧如下

df['status']=np.where(df.A.str.contains('|'.join(['a','b','c'])),'yes','no')
df.pivot(index='A', columns='status',values='A').fillna('').reset_index()
status    A   no  yes
0       bot       bot
1       cat       cat
2       dog  dog     
3       fly  fly