Python \"groupby\" 和 \"nunique\" [重复]

Question

我有一个 pandas 列 df.a = [6,6,8,10,10,6]，我使用 df.groupby('a).nunique() 并得到结果 df.a = [6,8,10 ].购买我预期的答案是 df.a = [6,8,10,6]。

请帮忙！谢谢。

Answer 1

这就是我认为您正在寻找的东西。

df = pd.DataFrame({"a":[6,6,8,10,10,6]})
df["b"] = df.a - df.a.shift(1).fillna(1)

df
Out[16]: 
    a    b
0   6  5.0
1   6  0.0
2   8  2.0
3  10  2.0
4  10  0.0
5   6 -4.0

然后取非零值并删除列 b

df = df[~(df.b==0)].drop(columns="b")

Out[19]: 
    a
0   6
2   8
3  10
5   6