Например, в dict1 все ключи 1, 2, 3 имеют одинаковое значение «a», но ключи 3 и 5 имеют разные значения, «b» и «d». Я хочу:
If N keys have the same value and N >=3, then I want to remove all other elements from the dict and only keep those N key values, which means 'b' & 'd' have to be removed from the dict.
Следующий код работает, но выглядит очень многословно. Есть лучший способ это сделать?
from collections import defaultdict
dict1 = {1:'a', 2:'a', '3':'b', '4': 'a', '5':'d'}
l1 = [1, 2, 3, 4, 5]
dict2 = defaultdict(list)
for k, v in dict1.items():
dict2[v].append(k)
to_be_removed = []
is_to_be_removed = False
for k, values in dict2.items():
majority = len(values)
if majority>=3:
is_to_be_removed = True
else:
to_be_removed.extend(values)
if is_to_be_removed:
for d in to_be_removed:
del dict1[d]
print(f'New dict: {dict1}')
Вы можете использовать collections.Counter
, чтобы получить частоту каждого значения, а затем использовать понимание словаря, чтобы сохранить только те ключи, которые имеют желаемое соответствующее значение:
from collections import Counter
dict1 = {1:'a', 2:'a', '3':'b', '4': 'a', '5':'d'}
ctr = Counter(dict1.values())
result = {key: value for key, value in dict1.items() if ctr[value] >= 3}
print(result)
Это выводит:
{1: 'a', 2: 'a', '4': 'a'}