Python Pandas 数据框搜索

Question

我在这里有代码，我在其中搜索 pandas 数据框上的系列。

                    df[(df['SlyWeekofFiscalYear']==wk) &
                       (df['IB']==bnd) &
                       (df['slyfiscalyear']==yr)]
                    ['Wholesale'].sum()

它在传递 kwarg bnd=None 的函数中。如果 bnd=None，有没有办法忽略第二行代码？

目前我有一个很长的 if 语句，但如果可能的话我想整理代码。

Answer 1

您可以将第二行更改为 ((bnd is None) | (df['IB'] == bnd))。如果 bnd 为 None，这将生成一个全真系列，并且由于您有 & 运算符，它不会对结果产生影响。