如何在pandas中的groupby（）。mean（）之后获取索引值列表？

Question

我坚持这个。我想从以下内容中获取name的列表，这是groupby().mean()与pandas DataFrame应用的结果。最具体地说，我想得到["John", "Mary", "Suzan", "Eric"]。

      score
name       
John   85.0
Mary   86.5
Suzan  90.0
Eric   100.0

以上结果是means，来自以下内容：

data = pandas.DataFrame({"name": names, "score": scores})
means = data.groupby("name").mean()

现在我有means，我想从'means' - ["John", "Mary", "Suzan", "Eric"]获取一个名单。这可以实现吗？

Answer 1

如果您有索引，请查看索引。如果您有MultiIndex，请参阅@ jezrael对get_level_values的回答。

means.index.tolist()

Answer 2

您可以使用：

print (list(means.index))
['John', 'Mary', 'Suzan', 'Eric']

另一个更好的解决方案是使用Series.unique并省略groupby：

print (data.name.unique())
['John' 'Mary' 'Suzan' 'Eric']