Вернуть счетчик объекта в многопроцессорную / картографическую функцию

У меня работает скрипт Python, который запускает одну и ту же функцию в нескольких потоках. Функции создают и обрабатывают 2 счетчика (c1 и c2). Результат всех счетчиков c1 из разветвленных процессов должен быть объединен вместе. То же самое относится ко всем счетчикам c2, возвращаемым разными вилками.

Мой (псевдо)-код выглядит так:

def countIt(cfg)
   c1 = Counter
   c2 = Counter
   #do some things and fill the counters by counting words in an text, like
   #c1= Counter({'apple': 3, 'banana': 0})
   #c2= Counter({'blue': 3, 'green': 0})    

   return c1, c2

if __name__ == '__main__':
        cP1 = Counter()
        cP2 = Counter()
        cfg = "myConfig"
        p = multiprocessing.Pool(4)  #creating 4 forks
        c1, c2 = p.map(countIt,cfg)[:2]
        # 1.) This will only work with [:2] which seams to be no good idea
        # 2.) at this point c1 and c2 are lists, not a counter anymore,
        # so the following will not work:
        cP1 + c1
        cP2 + c2

Следуя приведенному выше примеру, мне нужен результат, подобный: cP1 = Counter({'apple': 25, 'banana': 247, 'orange': 24}) cP2 = Counter({'red': 11, 'blue': 56, "зеленый": 3})

Итак, мой вопрос: как я могу подсчитать, что происходит с разветвленным процессом, чтобы объединить каждый счетчик (все c1 и все c2) в родительском процессе?

1 ответ

Решение

Вам нужно "разархивировать" свой результат, используя, например, цикл for-each. Вы получите список кортежей, где каждый кортеж представляет собой пару счетчиков: (c1, c2),
С вашим текущим решением вы фактически смешиваете их. Вы назначены [(c1a, c2a), (c1b, c2b)] в c1, c2 означающий, что c1 содержит (c1a, c2a) а также c2 содержит (c1b, c2b),

Попробуй это:

if __name__ == '__main__':
        from contextlib import closing

        cP1 = Counter()
        cP2 = Counter()

        # I hope you have an actual list of configs here, otherwise map will
        # will call `countIt` with the single characters of the string 'myConfig'
        cfg = "myConfig"

        # `contextlib.closing` makes sure the pool is closed after we're done.
        # In python3, Pool is itself a contextmanager and you don't need to
        # surround it with `closing` in order to be able to use it in the `with`
        # construct.
        # This approach, however, is compatible with both python2 and python3.
        with closing(multiprocessing.Pool(4)) as p:
            # Just counting, no need to order the results.
            # This might actually be a bit faster.
            for c1, c2 in p.imap_unordered(countIt, cfg):
                cP1 += c1
                cP2 += c2
Другие вопросы по тегам