Pandas：如何从 CSV 文件中读取特定行

Question

我有一个 csv 文件 example.csv -

     name  |  hits
   ---------------
     A    |  34
     B    |  30
     C    |  25
     D    |  20

在 Python 中使用 pandas ，如何只读取带有 hits > 20 的行？寻找类似的东西-

my_df = pd.read_csv('example.csv', where col('hits') > 20)

原文由 kev 发布，翻译遵循 CC BY-SA 4.0 许可协议

Answer 1

阅读整个 csv 并进行如下过滤

my_df =  pd.read_csv("example.csv")
my_df = my_df[my_df['hits']>20]

如果您在阅读时遇到内存问题，您可以设置 chunksize 参数以分块读取

原文由 Sociopath 发布，翻译遵循 CC BY-SA 4.0 许可协议

Answer 2

阅读整个 csv，然后使用 query() 方法选择所需的部分：

 required_df = my_df.query("hits > 20")

或者，

 required_df =df.loc[df['hits']>20]

原文由 Loochie 发布，翻译遵循 CC BY-SA 4.0 许可协议