DataFrame获得索引不同方法

当处理 DataFrame 或类似数据结构时，获得索引的方法有几种：

1. 使用布尔索引：对 DataFrame 使用布尔索引可以快速筛选出满足条件的行，并获取这些行的索引。

   index_list = df[df['A'] == 1].index.tolist()
   

2. 使用 .loc[] 或 .iloc[]：.loc[] 可以使用标签索引，.iloc[] 可以使用位置索引。你可以通过给定条件来筛选行，并获取这些行的索引。

   index_list = df.loc[df['A'] == 1].index.tolist()
   

   或者

   index_list = df.iloc[(df['A'] == 1).values].index.tolist()
   

3. 使用 .query() 方法：.query() 方法允许你使用类似 SQL 的查询语法来筛选行，并获取这些行的索引。

   index_list = df.query('A == 1').index.tolist()
   

这些方法都可以用于在 Pandas 中获取 DataFrame 中满足特定条件的行的索引。