python - Pandas DataFrame 查询 Python

Question

我正在尝试查询数据框的值。我的数据由 6 列 G-p1、G-p2、G-c、H-p1、H-p2、H-c 组成。所有列的值都是“左”或“右”，因为它们代表 parent / child 是左手还是右手基因型或惯用手。我想查询 parent 和 child 的惯用手的值。我试过了:

test1 = pd.DataFrame(data)
test1 = test1.query({
        'H-p1': 'left',
        'H-p2': 'left',
        'H-c': 'left'})
train_data = test1
predict_data = test1
model.fit(test1)
predict_data = predict_data.copy()
predict_data.drop('H-p1', axis=1, inplace=True)
predict_data.drop('H-p2', axis=1, inplace=True)
predict_data.drop('H-c', axis=1, inplace=True)
pred = model.predict_probability(predict_data)
print(pred.to_string())

但我收到此错误:

ValueError: expr must be a string to be evaluated, <class 'dict'> given

有什么建议？谢谢!

Answer 1

query 方法接收类似于您将用于 loc 过滤器的字符串表达式。
尝试这个:

test1 = test1.query("`H-p1` == 'left' and `H-p2` == 'left' and `H-c` == 'left'")
train_data = test1

反引号 ``用于指定列名。