Pandas 幕后发生了什么导致 MultiIndex 中的一个级别不被删除？

Question

考虑数据框df
请注意，columns 对象是单级 MultiIndex。

midx = pd.MultiIndex.from_product([list('AB')])
df = pd.DataFrame(1, range(3), midx)

   A  B
0  1  1
1  1  1
2  1  1

现在当我引用列 'A'

df.A

   A
0  1
1  1
2  1

我得到一个单列数据框，而不是我期望的系列对象。因此，我可以无限期地引用本专栏。

df.A.A.A.A.A

   A
0  1
1  1
2  1

作为另一个检查，我使用了xs

df.xs('A', axis=1)

   A
0  1
1  1
2  1

同样的问题。
pd.IndexSlice?

df.loc[:, pd.IndexSlice['A']]

   A
0  1
1  1
2  1

squeeze怎么样

df.A.squeeze()

0    1
1    1
2    1
Name: (A,), dtype: int64

这完全不是我所期望的。

1. 是什么阻止了它变成名为'A' 的系列对象？
2. 解决此问题最直观的方法是什么？
3. 有什么好的理由让我们想要一个单一级别的MultiIndex？

Answer 1

我写这个是为了解决这个问题。

def fix_single_level_multiindex(midx):
    return midx.get_level_values(0) if midx.nlevels == 1 else midx

或者

def fix_single_level_multiindex(midx):
    return midx.levels[0][midx.labels[0]] if midx.nlevels == 1 else midx