使用熊猫计算特定列中具有相同值的行

Question

我有一个csv文件，其中包含许多产品（40k）和许多带有价格，类别，名称的列... 我想计算每个类别中有多少行（产品）（直接从文件中获取）并显示它。 我如何使用熊猫来做到这一点？还是使用其他东西更好？

Answer 1

import pandas as pd

df = pd.read_csv(PATH_TO_CSV, usecols=['category','products'])
print(df.groupby(['category']).count())

第一行创建一个包含两列（类别和产品）的数据框，第二行打印出每个类别中的产品数量。