如何對一個字段進行不同計數，在 Pandas 中按另一個字段分組

Question

如果我想創建一個 dataframe，相當於 PANDAS 中的這個 SQL 怎么辦？

SELECT COUNTRY, COUNT(DISTINCT PRODUCT) AS UNIQUE_PRODUCTS 
FROM SALES
GROUP BY COUNTY

Answer 1

df = pd.DataFrame({
    'Country': ['A', 'A', 'B', 'B', 'B', 'C', 'C'],
    'Product': ['X', 'X', 'X', 'Y', 'Z', 'Y', 'Z']    
})

df.groupby('Country').Product.nunique()
>>>
Country
A    1
B    3
C    2
Name: Product, dtype: int64