如何根据python中值的某些部分按键分组?

Question

我有一个json文件与keys和Results。结果有三部分。这样的:

df.json()[0]['Results'][0]
{'categoryId': '2674',
'categoryName': 'software engineer',
'score': '1.0377672767639161'}

我想收集结果中具有相同categoryName的所有keys。然后数一数。这可能吗?

Answer 1

这样做的一种方法是遍历每一组:如果你只想知道一个组的计数，你可以这样做:

category_name = "software engineer"
count = 0
for cat in df.json()[0]['Results']:
if cat['categoryName'] == category_name:
count += 1

如果你想同时计算所有类别，你也可以这样做:

category_count = {}
for cat in df.json()[0]['Results']:
category = cat['categoryName']
if category in category_count.keys():
category_count[category] += 1
else:
category_count[category] = 1

来获取相应的键，你可以这样做:

category_count = {}
for key, cat in enumerate(df.json()[0]['Results']):
category = cat['categoryName']
if category in category_count.keys():
category_count[category].append(key)
else:
category_count[category] = [key]

这样你就可以像这样得到所有的categoryName="software engineer"键:category_count["software_engineer"]