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

Question

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

标签： pythonpandascsvcountrows

Answer 1

import pandas as pd

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

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