DataFrame Выберите записи, которые находятся в списке

df = pd.DataFrame({'A': [5,6,3,4], 'B': [1,2,3,5]})
df[df['A'].isin([3, 6])]
Real Raccoon