将数据帧中的值与列表值进行比较后保存该值

Question

我试图将列表中的值与dataframe列的值进行比较。如果两者相等，我想保存数据帧的整行。 我无法编写保存数据帧整行的指令

以下是dataframe和list的一些数据：

print(approval_polls.head(5))

 start_date    end_date         pollster    sponsor  sample_size population  \
0  2020-02-02  2020-02-04           YouGov  Economist       1500.0          a   
1  2020-02-02  2020-02-04           YouGov  Economist        376.0          a   
2  2020-02-02  2020-02-04           YouGov  Economist        523.0          a   
3  2020-02-02  2020-02-04           YouGov  Economist        599.0          a   
4  2020-02-07  2020-02-09  Morning Consult        NaN       2200.0          a   


excel_doc = ['Monmouth University' 'Selzer & Co.' 'ABC News/The Washington Post'
 'Siena College/The New York Times Upshot' 'YouGov']

我开始编写的代码如下：

approval_polls = approval_polls[approval_polls['pollster'].isin(excel_doc)]

我得到的结果不对

print (approval_polls)

[start_date, end_date, pollster, sponsor, sample_size, population, ...]

这里怎么了

谢谢你的建议

Answer 1

是的，isin是一条路要走

excel_doc = ['Monmouth University' 'Selzer & Co.' 'ABC News/The Washington Post'
             'Siena College/The New York Times Upshot', 'YouGov']

df = df[df['pollster'].isin(excel_doc)]