python - 从多索引数据框中获取一个索引

Question

如果我有一个多索引数据框，我怎样才能只在一个索引中获取信息？

如果我的 df 是这样的:

first  second
bar    one       0.469112
       two      -0.282863
baz    one      -1.509059
       two      -1.135632
foo    one       1.212112
       two      -0.173215
qux    one       0.119209
       two      -1.044236

我想要索引 first 中的值列表。我习惯于做这样的事情:

df.index.tolist()

返回:

['bar','baz','foo','qux']

Answer 1

您可以使用 pd.Index.get_level_values .这是一个演示:

df = pd.DataFrame([['A', 'B', 1], ['A', 'C', 2], ['X', 'Y', 3], ['X', 'Z', '4']],
                  columns=['idx1', 'idx2', 'value'])

df = df.set_index(['idx1', 'idx2'])

res = df.index.get_level_values(0).tolist()

['A', 'A', 'X', 'X']