根据 id 访问 groupby 值？

Question

我有一个包含 ID 和重量信息的数组，我想根据关于最小和最大重量的 ID 对数组进行分组：

a= pd.Series(weight).groupby(id).agg(["min", "max"])

我想根据 id 访问每个最小/最大元素。我的问题是如何根据键（id）保存和访问每个 min/max 值，以便我可以访问每个元素？

我使用 w_min=a.iloc[0,0] 访问每个元素，访问每个元素的最佳方式是什么？

Answer 1

如果我没理解错的话，你可以使用 .to_dict() 然后你可以通过按键访问你的值（在本例中 id）：


#... code as in your question

out = a.to_dict(orient="index")

print(out)
print(out[5]["min"])  # <-- access by `5` and `min`

打印：

{
    1: {"min": 100, "max": 200},
    3: {"min": 258, "max": 585},
    5: {"min": 89, "max": 632},
}
89

Accessing groupby value based on id?