CODE EXAMPLE FOR PYTHON

pandas groupby agg count unique

df.groupby("A").agg({"B": lambda x: x.nunique()})