日々精進

新しく学んだことを書き留めていきます

pandasでmulti indexになっているときにindexの列で絞り込む

以下のようにget_level_valuesメソッドでindexのうち一つの列を選択してmaskをかけられる。

        mask: Series = self.features.index.get_level_values(1) == pred_date
        self.features.loc[mask, SALES_COUNT] = pred[PRED_SALES_COUNT]

multi indexはほんとは扱いにくいから使わないようにしたい。。

参考:

stackoverflow.com