復(fù)雜的標(biāo)題，但很簡單的問題。我有一個帶有 MultiIndex 的 DataFrame：我想要上面框架的行，但'Filter Column'必須大于或等于filter_value下面系列中的值。filter_value = Series([1, 3], ['red', 'blue'])這個玩具問題的正確解決方案是相同的數(shù)據(jù)幀，但只剩下(red, 2),(blue, 2)和(blue, 3)行。要設(shè)置上述框架：arrays = [['red', 'red', 'blue', 'blue', 'blue'], [1, 2, 1, 2, 3]]idx = MultiIndex.from_arrays(arrays, names=['Color', 'Count'])values = Series(2, idx, name='Value')ratios = Series(range(5), idx, name='Filter Column')df = concat([values, ratios], axis='columns')
2 回答
回首憶惘然
TA貢獻(xiàn)1847條經(jīng)驗(yàn) 獲得超11個贊
看起來你需要get_level_values+ map，然后使用值比較得到 df 的布爾過濾器
df[df['Filter Column'].values>=df.index.get_level_values(0).map(filter_value)]
Out[108]:
Value Filter Column
Color Count
red 2 2 1
blue 2 2 3
3 2 4
猛跑小豬
TA貢獻(xiàn)1858條經(jīng)驗(yàn) 獲得超8個贊
你可以試試這個：
pd.concat(df.align(filter_value.rename('filter'), level=0, axis=0), axis=1)\
.loc[lambda x: x['Filter Column']>=x['filter']]
輸出：
Value Filter Column filter
Color Count
red 2 2 1 1
blue 2 2 3 3
3 2 4 3
添加回答
舉報
0/150
提交
取消