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

Question

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

Answer 1

import pandas as pd

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

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