Python: Đếm tần số của các phần tử trong danh sách các danh sách

Question

Thực ra, tôi có tập dữ liệu về "cuộc họp". Ví dụ, A, B, C có một cuộc họp, sau đó danh sách sẽ là [A, B, C]. Như thế này, mỗi danh sách sẽ chứa một danh sách các thành viên tham gia cuộc họp. Do đó:

line1 = (A, B, C)

dòng2 = (A, C, D, E)

Line3 = (D, F, G)

..

Tôi chỉ muốn đếm số lần mỗi cặp thành viên gặp nhau. Ví dụ, thành viên A gặp C hai lần từ line1 và line2 và thành viên B gặp C một lần từ dòng1. Vì vậy, tôi muốn thực hiện một biểu đồ như thế này ..

A B C D E F G... 

A . 1 2 1 ... 

B 1 . 1 0 

C 

...

Tôi nghĩ rằng nó sẽ được dễ dàng ở người đầu tiên nhưng tôi khá bối rối. Xin hãy giúp tôi và cảm ơn bạn rất nhiều trước.

Answer 1

Đây là vấn đề cấu trúc dữ liệu khá đơn giản với mảng 2D hoặc dict. Mảng hiệu quả hơn nếu bạn có nhiều người, nhưng tôi sẽ giả sử bạn không làm như vậy.

times_met = defaultdict(int) 
for line in lines: 
    for pair in itertools.combinations(line, 2) 
     times_met[pair] += 1 

# How many times person a meets person b is described by the following (s.t. a < b) 
print times_met[(a, b)] 

Lưu ý rằng điều này thực sự không hiệu quả nếu bạn có các cuộc họp lớn và các thuật toán hiệu quả hơn có thể tồn tại.

Answer 2

Có vẻ như bạn sẽ có thể giải quyết vấn đề này với việc bổ sung ma trận. Nếu bạn biết tổng số người (G trong câu hỏi), thì câu trả lời của bạn sẽ là ma trận GxG. Tạo một ma trận với các kết hợp từ line1 GxG, sau đó thêm vào một ma trận với các kết hợp từ dòng2 GxG vv

Answer 3

Thay vì tần số bằng tay tổng hợp, sử dụng collections.counter cùng với itertools:

from collections import Counter 
from itertools import chain, combinations 

meets = Counter(chain.from_iterable(combinations(line, 2) for line in lines)) 

đâu lines là một iterable của iterables của tên.