Q похоже на это: используйте список значений для выбора строк из pandas фрейма данных
Я хочу, чтобы кадр данных, если значение в двух столбцах находится в списке. Верните оба столбца (объедините результаты # 1 и # 4.
import numpy as np
from pandas import *
d = {'one' : [1., 2., 3., 4] ,'two' : [5., 6., 7., 8.],'three' : [9., 16., 17., 18.]}
df = DataFrame(d)
print df
checkList = [1,7]
print df[df.one == 1 ]#1
print df[df.one == 7 ]#2
print df[df.two == 1 ]#3
print df[df.two == 7 ]#4
#print df[df.one == 1 or df.two ==7]
print df[df.one.isin(checkList)]