Skip to main content
 首页 » 编程设计

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

2025年05月04日13jackei

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

如果我的 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']