Skip to main content
 首页 » 编程设计

python之有没有更pythonic的方式来写这个

2025年05月04日15soundcode

我想让它更像 pythonic。

user_df[-1:]['status_id'].values[0] in {3,5} 

我最初尝试了 user_id.ix[-1:, 'status_id'].isin([3,5]),但没有成功。

有什么建议吗?顶级版本有效,但看起来有点奇怪。

请您参考如下方法:

你可以试试:

user_id['status_id'].iloc[-1:].isin([3,5]) 

示例:

user_id = pd.DataFrame({'status_id':[1,2,3]}) 
print (user_id) 
   status_id 
0          1 
1          2 
2          3 
 
#iloc without : [-1] return scalar 
print (user_id['status_id'].iloc[-1] in set({3,5})) 
True 
 
#iloc with : [-1:] return vector - Series 
print (user_id['status_id'].iloc[-1:].isin([3,5])) 
2    True 
Name: status_id, dtype: bool