从 Pandas 数据框的 MultiIndex 中获取特定索引作为列表

Question

我想以列表形式获取 Pandas 数据框的特定索引MultiIndex。鉴于这个例子import pandas as pd; import numpy as npnp.random.seed(42)df = pd.DataFrame(np.random.randint(5, size=(5, 4)), columns=list('ABCD'))df.set_index(['A', 'B'], inplace=True)其中df定义为     C  DA B3 4  2  44 1  2  22 4  3  24 1  3  13 4  0  3我想提取[4 1 4 1 4]对应于第二个索引B。如何才能做到这一点？

Answer 1

使用 get_level_values

df.index.get_level_values(level=1).tolist()

Out[1040]: [4, 1, 4, 1, 4]

要么 reset_index

df.reset_index(level=1).B.tolist()

Out[1041]: [4, 1, 4, 1, 4]

Answer 2

此解决方案使用堆叠np.stack()和切片：

np.stack(df.index.values, axis=0)[:,1]