如果我有一个多索引数据框,我怎样才能只在一个索引中获取信息?
如果我的 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']
请您参考如下方法:
您可以使用 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']
