python - Pandas :以编程方式选择多列

Question

现在，我选择多行，其中某些列以这种方式具有特定值:

df.loc[(df['col1'] == val1) & (df['col2'] == val2)]

有没有办法以编程方式执行此操作，我将列键/值作为字典提供？像这样:

def get_df(cols)

   df.loc[ (df[k] == v) for k,v in cols.items() ]

但我不确定如何“与”表达式。有什么想法吗？

Answer 1

您可以从字典中创建一个 Series 并进行比较:

import numpy as np
import pandas as pd
np.random.seed(0)

df = pd.DataFrame(np.random.randint(0, 5, (100, 3)), columns = list("ABC"))
cols = {"A": 0, "B": 3, "C": 3}

df[(df == pd.Series(cols)).all(axis=1)]
Out: 
    A  B  C
94  0  3  3

或者，使用 np.logical_and与 reduce :

df[np.logical_and.reduce([(df[k] == v) for k,v in cols.items()])]

    A  B  C
94  0  3  3