pandas选择列名不同方式

119 阅读1分钟

在 Pandas 中,df[df.A == 1].index.tolist()df[df["A"]== 1].index.tolist() 具有相同的效果。它们的区别在于选择列名的方式不同:

  1. df.A:使用属性方式选择列名。这种方式适用于列名中没有空格或特殊字符的情况。

  2. df["A"]:使用字典索引方式选择列名。这种方式适用于任何列名,包括有空格或特殊字符的情况。

因此,如果你的列名中没有空格或特殊字符,那么这两种写法是等价的。如果列名中有空格或特殊字符,那么你必须使用 df["A"] 的方式。

综上所述,df[df.A == 1].index.tolist()df[df["A"]== 1].index.tolist() 是等价的,它们都会获取 DataFrame 中 'A' 列值等于 1 的所有行的索引,并以列表的形式返回。