如何在蟒蛇系列

Question

我有一个数据帧过滤的价值观和我保存这个数据帧的第一行如下

first_row = data_model.loc[0,:] 

如果我打印出first_row，它看起来像

label    class 
wa      3 
not     0 
im      2 
time     0 
see     2 
like     0 
going     2 
amp     1 
get     1 
one     0 

我只是想保留值大于0，所以我用first_row[first_row > 0]，但我得到一个错误信息，如：

TypeError: '>' not supported between instances of 'str' and 'int'.

如果我使用first_row[first_row != 0]，我会得到我想要的。

Answer 1

使用数据框的查询功能将返回所需的结果

df=pd.DataFrame({"label:["wa","not","im","time","see","lke","going"],"amp":`[3,0,2,0,2,0,2]}) 

df=df.query("amp>0") 

print(df) 

      amp label 
     0 3  wa 
     2 2  im 
     4 2 see 
     6 2 going