根据条件python计数项目数量的最佳方法

Question

嗨，大家好，我有多种条件可以检查一系列对象：

我的示例数组是这样的

[{"id":1 , "type":"viw"},{"id":1 , "type":"edit"},{"id":1 , "type":"crt"}]

，因为这些记录将在10000中，我想以最好的方式计算这些记录的总和。

目前这样做，

   view_count = [element for element in user_record if element['type'] == "viw"  ]
   edit_count = [element for element in user_record if element['type'] == "edit"]
   crt_count = [element for element in user_record if element['type'] =="crt"]

   and then len(crt_count) ,len(edit_count) ,len(view_count) 

对我来说似乎有点贵。请指南我可以优化的方式吗？

我可以在此列表中使用第二个问题或条件吗？

Answer 1

使用：

d={}
for dct in sample_arr:
    key = dct['type']
    if d.get(key) is None:
        d[key] = 0
    d[key] += 1

dict(d)
 {'viw': 1, 'edit': 1, 'crt': 1}

或使用defaultDict：

from collections import defaultdict  
e = defaultdict(int)
for dct in sample_arr:
    e[dct['type']] +=1
dict(e)
 {'viw': 1, 'edit': 1, 'crt': 1}

Answer 2

来计算其值

result = {}
for element in user_record:
   try:
        result[element["type"]] += 1
   except KeyError: 
        result[element["type"]] = 1

只需将每个“类型”收集到字典键中，然后用@not机器人的评论 ：
“基本上相同想法的另一种方法：”

collections.Counter(map(operator.itemgetter('type'), my_list))

第二个问题：答案是是，您可以使用或这样的方法：

edit_count = [element for element in user_record if element['type'] == "edit" or element['type'] == 'viw']

Answer 3

如果您关心避免多次迭代，只需通过一次迭代来计算：

sample_arr = [{"id":1 , "type":"viw"},{"id":1 , "type":"edit"},{"id":1 , "type":"crt"}]

view_count, edit_count, crt_count = 0, 0, 0

for d in sample_arr:
    if d['type'] == 'viw':
        view_count += 1
    elif d['type'] == 'edit':
        edit_count += 1
    elif d['type'] == 'crt':
        crt_count += 1

Answer 4

不使用循环的另一个选项（但使用额外的内存）：

import pandas as pd

arr = [{"id":1 , "type":"viw"},{"id":1 , "type":"edit"},{"id":1 , "type":"crt"},
       {"id":2 , "type":"viw"},{"id":2 , "type":"edit"},{"id":2 , "type":"crt"}]

df = pd.DataFrame(arr)
res = df['type'].value_counts().to_dict()

>>> res
{'viw': 2, 'edit': 2, 'crt': 2}