python - 将多列与多索引列 Pandas 中的同一列相乘

Question

我有这个数据框

col = pd.MultiIndex.from_arrays([['Jan', 'Jan', 'Jan', 'Cost'],
                                ['WK1', 'WK2', 'WK3', 'Cost']])
data = pd.DataFrame(np.random.randn(4, 4), columns=col)
data

我想用成本乘以 1 月的所有列，所以我这样尝试

data[['Jan']] * data[['Cost']]

但它用 Nan 填充了我的值。我有办法做到这一点吗？

Answer 1

我认为您需要删除 [] 以使用 mul 选择 DataFrame :

print (data['one'].mul(data['two']))

或者:

print (data['one'] * data['two'])

也可以使用函数 xs :

print (data.xs('one', axis=1, level=0) * data.xs('two', axis=1, level=0))