时间戳

首页 > 解决方案 > pandas df 基于多个条件的查找值

pandas - pandas df 基于多个条件的查找值

问题描述

我的 df 称为符号如下所示：

                                   local_symbol            code globex expiry_date type  strike
symbol                                                                                         
OZN20121221P00014900_FOP  P OZN DEC 12    14900  CBT_21_F2013_S    OZN  2012-12-21    P   149.0
OZN20121221C00012500_FOP  C OZN DEC 12    12500  CBT_21_F2013_S    OZN  2012-12-21    C   125.0
OZN20121221P00012450_FOP  P OZN DEC 12    12450  CBT_21_F2013_S    OZN  2012-12-21    P   124.5
OZN20121221C00013900_FOP  C OZN DEC 12    13900  CBT_21_F2013_S    OZN  2012-12-21    C   139.0
OZN20121221C00010700_FOP  C OZN DEC 12    10700  CBT_21_F2013_S    OZN  2012-12-21    C   107.0

使用 pandas 0.22.0 以下工作：

exp_date = dt.date(2012, 12, 21)
code = 'CBT_21_F2013_S'
type = 'P'
strike = 124.5
symbols.loc[(symbols.expiry_date == exp_date)
            & (symbols.code == code)
            & (symbols.type == type)
            & (symbols.strike == strike)]

并返回 OZN20121221P00012450_FOP 作为预期值。在 Pandas 0.23 中，我得到一个空数据框。您的帮助将不胜感激。

标签： pandaspython-3.6

解决方案

我相信需要将 column 转换expiry_date为dates 并比较浮动使用numpy.isclose：

symbols.loc[(symbols.expiry_date.dt.date == exp_date)
            & (symbols.code == code)
            & (symbols.type == type)
            & np.isclose((symbols.strike, strike)]

推荐阅读