在pandas数据框中的任意位置搜索值

2024-04-18 21:24:45 发布

您现在位置:Python中文网/ 问答频道 /正文

这似乎是一个简单的问题,但我找不到它之前提出过(thisthis很接近,但答案不是很好)。

问题是:如果我想在df(我不知道它在哪一列)中的某处搜索值,并返回所有匹配的行。

最平庸的方法是什么?有什么比这更好的吗

for col in list(df):
    try:    
        df[col] == var
        return df[df[col] == var]
    except TypeError:
        continue 

是吗?


Tags: 方法答案indfforreturnvarcol
3条回答

您可以尝试以下代码:

import pandas as pd
x = pd.read_csv(r"filePath")
x.columns = x.columns.str.lower().str.replace(' ', '_')
y = x.columns.values
z = y.tolist()
print("Note: It take Case Sensitive Values.")
keyWord = input("Type a Keyword to Search: ")
try:
    for k in range(len(z)-1):
        l = x[x[z[k]].str.match(keyWord)]
        print(l.head(10))
        k = k+1
except:
    print("")

可以对整个数据帧执行相等比较:

df[df.eq(var1).any(1)]

另一个选项是使用numpy比较:

df[(df.values.ravel() == var1).reshape(df.shape).any(1)]

您应该使用isin,这是返回列,是要行冷检查'答案:-)

df.isin(['bal1']).any()
A        False
B         True
C        False
CLASS    False
dtype: bool

或者

df[df.isin(['bal1'])].stack() # level 0 index is row index , level 1 index is columns which contain that value 
0  B    bal1
1  B    bal1
dtype: object

相关问题 更多 >