在列中顯示特定值的唯一計數，使用 pandas 按附加列分組

Question

我有一個數據集 df，我想按一列分組，顯示每個唯一值的計數並顯示正確的列。

數據

Id      location
e-db    ny
e-db    ny
e-db    ny
f-a     ny
f-a     ny
gr-x    ca

想要的

Id      location    count
e-db    ny          3
f-a     ny          2
gr-x    ca          1

正在做

df.groupby('location')['Id'].nunique()

但是，這並沒有向我顯示唯一 ID 和實際值。 任何建議表示贊賞

Answer 1

你也可以這樣做：

df2 = pd.DataFrame()
df2['location'] = df.groupby('Id')['location'].nth(0)
df2['count'] = df.groupby('Id').count()

Output：

Id  location    count
e-db    ny       3
f-a     ny       2
gr-x    ca       1