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

Question

我想让它更像 pythonic。

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

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

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

Answer 1

你可以试试:

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