python 循环筛选数据

Question

因为这个文件很大。我想根据上面的第一列的字符串进行筛选，而且想要保留的数据也保存为set（）如下
events = set()#要保留的数据如下

还是初学者，所以很多不懂。自己思路大概是这样

for rows in tr: #遍历list(tr)每行数据
   for i in events:
     if (tr[rows[0]] == event[i])： # tr[rows[0]]可以索引至第一张图的第一列数据？好像set（）不支持这样索引
             #保存

不知道怎么实现，想请教一下感谢

Answer 1

使用 pandas 库读取、筛选数据方便多了，以下是 ipython 示例

In [1]: import pandas as pd

In [2]: pd.read_csv('a.csv')
Out[2]: 
     a  b
0  one  1
1  two  2
2  one  3

In [3]: df = _

In [4]: df[df.a.eq('one')]
Out[4]: 
     a  b
0  one  1
2  one  3

参考

pandas, https://pandas.pydata.org/