python - dataframe.loc，索引上的布尔值

Question

我目前有一个数据框：

df = |0 10|
     |1 20|
     |2 30|

我正在尝试返回一个新的数据帧df2 = df['INDEXNAME' < 2]，以便它只返回前两行。

我试过了：

df.loc[df<100]
df.loc[df['INDEXNAME']<100]

但这些给了我一个

  

KeyError：'INDEXNAME'

有什么想法？谢谢！

Answer 1

您可以使用查询：

df= pd.DataFrame({'value':[10,20,30]}, index = [20,2,0])
df.index.name = "beta"

df.query("beta < 2")
#    value
#beta   
#0     30

或：

df[df.index < 2]
# value
#0   30