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

Question

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

Answer 1

import pandas as pd

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

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