如何在熊猫中实施这种搜索？

Question

假设我有一个数据框，名称很多，例如：

[[jack,rose,mike],
 [mike,jack,lee],
 [jeff,jack,alex]]

我需要的是一个函数，当我输入“ jack”时，返回的数据帧就像：

[[1,0,0],
[0,1,0],
[0,1,0]]

Pandas中是否有符合我要求的方法？

Answer 1

您可能想要eq，它是“等于”的缩写

import pandas as pd

df = pd.DataFrame([['jack', 'rose', 'mike'],
                   ['mike', 'jack', 'lee'],
                   ['jeff', 'jack', 'alex']],
                  columns=['a','b','c'])

df.eq('jack') * 1

Answer 2

您可以直接将DataFrame项与标量进行比较：

(df == 'jack').astype(int)