python - 使用 Pandas 计算特定列中具有相同值的行

Question

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

Answer 1

import pandas as pd

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

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