Поиск количества кортежей с одинаковым первым и третьим элементом в списке кортежей - PullRequest
Новая
       23

Поиск количества кортежей с одинаковым первым и третьим элементом в списке кортежей

0 голосов
/ 28 июня 2018

У меня есть список кортежей, каждый из которых состоит из трех элементов: 

z = [(1, 4, 2015), (1, 11, 2015), (1, 18, 2015), (1, 25, 2015), (2, 1, 2015), (2, 8, 2015), (2, 15, 2015), (2, 22, 2015), (3, 1, 2015), (3, 8, 2015), (3, 15, 2015), (3, 22, 2015), (3, 29, 2015), (4, 5, 2015), (4, 12, 2015), (4, 19, 2015), (4, 26, 2015), (5, 3, 2015), (5, 10, 2015), (5, 17, 2015), (5, 24, 2015), (5, 31, 2015), (6, 7, 2015), (6, 14, 2015), (6, 21, 2015), (6, 28, 2015), (7, 5, 2015), (7, 12, 2015), (7, 19, 2015), (7, 26, 2015), (8, 2, 2015), (8, 9, 2015), (8, 16, 2015), (8, 23, 2015), (8, 30, 2015), (9, 6, 2015), (9, 13, 2015), (9, 20, 2015), (9, 27, 2015), (10, 4, 2015), (10, 11, 2015), (10, 18, 2015), (10, 25, 2015), (11, 1, 2015), (11, 8, 2015), (11, 15, 2015), (11, 22, 2015), (11, 29, 2015), (12, 6, 2015), (12, 13, 2015), (12, 20, 2015), (12, 27, 2015), (1, 3, 2016), (1, 10, 2016), (1, 17, 2016), (1, 24, 2016), (1, 31, 2016)]

Я хочу найти количество кортежей в списке с одинаковыми первым и третьим элементами, как и в первом пункте 1 и третьем элементе 2015 года, имеется 4 кортежа; с первым пунктом 2 и третьим пунктом 2015, есть 4 кортежа.

Я пытался: 

for tup in z:
    a=tup[0]
    b=tup[2]
    print(len(set({a:b})))

Не дает желаемого результата. Как это сделать?

Ответы [ 7 ]

0 голосов
/ 28 июня 2018

from collections import Counter tmp = [(x[0],x[2]) for x in z] print(Counter(tmp))

вывод придет как Counter({(5, 2015): 5, (11, 2015): 5, (8, 2015): 5, (3, 2015): 5, (1, 2016): 5, (10, 2015): 4, (2, 2015): 4, (6, 2015): 4, (12, 2015): 4, (7, 2015): 4, (9, 2015): 4, (4, 2015): 4, (1, 2015): 4})

0 голосов
/ 28 июня 2018

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

z = [(1, 4, 2015), (1, 11, 2015), (1, 18, 2015), (1, 25, 2015), (2, 1, 2015), (2, 8, 2015), (2, 15, 2015), (2, 22, 2015), (3, 1, 2015), (3, 8, 2015), (3, 15, 2015), (3, 22, 2015), (3, 29, 2015), (4, 5, 2015), (4, 12, 2015), (4, 19, 2015), (4, 26, 2015), (5, 3, 2015), (5, 10, 2015), (5, 17, 2015), (5, 24, 2015), (5, 31, 2015), (6, 7, 2015), (6, 14, 2015), (6, 21, 2015), (6, 28, 2015), (7, 5, 2015), (7, 12, 2015), (7, 19, 2015), (7, 26, 2015), (8, 2, 2015), (8, 9, 2015), (8, 16, 2015), (8, 23, 2015), (8, 30, 2015), (9, 6, 2015), (9, 13, 2015), (9, 20, 2015), (9, 27, 2015), (10, 4, 2015), (10, 11, 2015), (10, 18, 2015), (10, 25, 2015), (11, 1, 2015), (11, 8, 2015), (11, 15, 2015), (11, 22, 2015), (11, 29, 2015), (12, 6, 2015), (12, 13, 2015), (12, 20, 2015), (12, 27, 2015), (1, 3, 2016), (1, 10, 2016), (1, 17, 2016), (1, 24, 2016), (1, 31, 2016)]
newz = [(i[0],i[-1]) for i in z]
for i in list(set(newz)):
   print(str(i)+' '+str(newz.count(i)))

Выход: 

(10, 2015) 4
(5, 2015) 5
(2, 2015) 4
(11, 2015) 5
(6, 2015) 4
(8, 2015) 5
(3, 2015) 5
(12, 2015) 4
(7, 2015) 4
(9, 2015) 4
(1, 2016) 5
(4, 2015) 4
(1, 2015) 4
0 голосов
/ 28 июня 2018

Вы можете сохранить счет в диктовке, снабженной кортежем, состоящим из первого и третьего элемента из исходного списка кортежей, например ::

import collections

z = [(1, 4, 2015), (1, 11, 2015), (1, 18, 2015), (1, 25, 2015), (2, 1, 2015), (2, 8, 2015),
     (2, 15, 2015), (2, 22, 2015), (3, 1, 2015), (3, 8, 2015), (3, 15, 2015), (3, 22, 2015),
     (3, 29, 2015), (4, 5, 2015), (4, 12, 2015), (4, 19, 2015), (4, 26, 2015), (5, 3, 2015),
     (5, 10, 2015), (5, 17, 2015), (5, 24, 2015), (5, 31, 2015), (6, 7, 2015), (6, 14, 2015),
     (6, 21, 2015), (6, 28, 2015), (7, 5, 2015), (7, 12, 2015), (7, 19, 2015), (7, 26, 2015),
     (8, 2, 2015), (8, 9, 2015), (8, 16, 2015), (8, 23, 2015), (8, 30, 2015), (9, 6, 2015),
     (9, 13, 2015), (9, 20, 2015), (9, 27, 2015), (10, 4, 2015), (10, 11, 2015),
     (10, 18, 2015), (10, 25, 2015), (11, 1, 2015), (11, 8, 2015), (11, 15, 2015),
     (11, 22, 2015), (11, 29, 2015), (12, 6, 2015), (12, 13, 2015), (12, 20, 2015),
     (12, 27, 2015), (1, 3, 2016), (1, 10, 2016), (1, 17, 2016), (1, 24, 2016), (1, 31, 2016)]

counter = collections.defaultdict(int)  # Use a dict factory to save some time
for element in z:  # iterate over the tuples
    counter[(element[0], element[2])] += 1  # increase the count for each match

# finally, lets print the results
for k, count in counter.items():
    print("{}: {}".format(k, count))

Что даст вам: 

(1, 2015): 4
(2, 2015): 4
(3, 2015): 5
(4, 2015): 4
(5, 2015): 5
(6, 2015): 4
(7, 2015): 4
(8, 2015): 5
(9, 2015): 4
(10, 2015): 4
(11, 2015): 5
(12, 2015): 4
(1, 2016): 5
0 голосов
/ 28 июня 2018

Использование collections.Counter с operator.itemgetter: 

from collections import Counter
from operator import itemgetter

res = Counter(map(itemgetter(0, 2), z))

print(res)

Counter({(1, 2015): 4,
         (1, 2016): 5,
         (2, 2015): 4,
         (3, 2015): 5,
         (4, 2015): 4,
         (5, 2015): 5,
         (6, 2015): 4,
         (7, 2015): 4,
         (8, 2015): 5,
         (9, 2015): 4,
         (10, 2015): 4,
         (11, 2015): 5,
         (12, 2015): 4})
0 голосов
/ 28 июня 2018

используя стандартный питон itertools.groupby: 

from itertools import groupby

for grp, elmts in groupby(z, lambda x: (x[0], x[2])):
    print(grp, len(list(elmts)))

Edit:

еще более приятное решение с использованием operator.itemgetter вместо lambda: 

from operator import itemgetter
from itertools import groupby

for grp, elmts in groupby(z, itemgetter(0, 2)):
    print(grp, len(list(elmts)))

Выход: 

(1, 2015) 4
(2, 2015) 4
(3, 2015) 5
(4, 2015) 4
(5, 2015) 5
(6, 2015) 4
(7, 2015) 4
(8, 2015) 5
(9, 2015) 4
(10, 2015) 4
(11, 2015) 5
(12, 2015) 4
(1, 2016) 5
0 голосов
/ 28 июня 2018

Использование collections.

Ex: 

import collections
d = collections.defaultdict(int)
z = [(1, 4, 2015), (1, 11, 2015), (1, 18, 2015), (1, 25, 2015), (2, 1, 2015), (2, 8, 2015), (2, 15, 2015), (2, 22, 2015), (3, 1, 2015), (3, 8, 2015), (3, 15, 2015), (3, 22, 2015), (3, 29, 2015), (4, 5, 2015), (4, 12, 2015), (4, 19, 2015), (4, 26, 2015), (5, 3, 2015), (5, 10, 2015), (5, 17, 2015), (5, 24, 2015), (5, 31, 2015), (6, 7, 2015), (6, 14, 2015), (6, 21, 2015), (6, 28, 2015), (7, 5, 2015), (7, 12, 2015), (7, 19, 2015), (7, 26, 2015), (8, 2, 2015), (8, 9, 2015), (8, 16, 2015), (8, 23, 2015), (8, 30, 2015), (9, 6, 2015), (9, 13, 2015), (9, 20, 2015), (9, 27, 2015), (10, 4, 2015), (10, 11, 2015), (10, 18, 2015), (10, 25, 2015), (11, 1, 2015), (11, 8, 2015), (11, 15, 2015), (11, 22, 2015), (11, 29, 2015), (12, 6, 2015), (12, 13, 2015), (12, 20, 2015), (12, 27, 2015), (1, 3, 2016), (1, 10, 2016), (1, 17, 2016), (1, 24, 2016), (1, 31, 2016)]
for i in z:
    d[(i[0], i[2])] += 1
print(d)

Выход: 

defaultdict(<type 'int'>, {(10, 2015): 4, (5, 2015): 5, (2, 2015): 4, (11, 2015): 5, (6, 2015): 4, (8, 2015): 5, (3, 2015): 5, (12, 2015): 4, (7, 2015): 4, (9, 2015): 4, (4, 2015): 4, (1, 2016): 5, (1, 2015): 4})
0 голосов
/ 28 июня 2018

В чистом питоне используйте Counter с генератором, спасибо @Felix: 

from collections import Counter

out = Counter((x[0], x[2]) for x in z)
print (out)
Counter({(3, 2015): 5, 
         (5, 2015): 5, 
         (8, 2015): 5,
         (11, 2015): 5, 
         (1, 2016): 5, 
         (1, 2015): 4, 
         (2, 2015): 4, 
         (4, 2015): 4, 
         (6, 2015): 4, 
         (7, 2015): 4, 
         (9, 2015): 4, 
         (10, 2015): 4,
         (12, 2015): 4})

В совокупности панд подсчитывает GroupBy.size, вывод составляет Series: 

s = pd.DataFrame(z).groupby([0,2]).size()
print (s)
0   2   
1   2015    4
    2016    5
2   2015    4
3   2015    5
4   2015    4
5   2015    5
6   2015    4
7   2015    4
8   2015    5
9   2015    4
10  2015    4
11  2015    5
12  2015    4
dtype: int64
...