def count_elements(data): length=0 for elt in data: length+=len(data[elt]) return length def copyData(s, d): d = {} for i in s: d[i]=[] for obs in s[i]: d[i].append(obs) def r_solution(data, K): return data