在 Pandas 中,df[df.A == 1].index.tolist() 和 df[df["A"]== 1].index.tolist() 具有相同的效果。它们的区别在于选择列名的方式不同:
-
df.A:使用属性方式选择列名。这种方式适用于列名中没有空格或特殊字符的情况。 -
df["A"]:使用字典索引方式选择列名。这种方式适用于任何列名,包括有空格或特殊字符的情况。
因此,如果你的列名中没有空格或特殊字符,那么这两种写法是等价的。如果列名中有空格或特殊字符,那么你必须使用 df["A"] 的方式。
综上所述,df[df.A == 1].index.tolist() 和 df[df["A"]== 1].index.tolist() 是等价的,它们都会获取 DataFrame 中 'A' 列值等于 1 的所有行的索引,并以列表的形式返回。