Question

在下面创建几行以迭代和求和字典中的每种可能的元素组合。例如，如果字典的长度为5，则我希望任意2个元素，任意3个元素，任意4个元素的总和。

import itertools

di = {'a': 1, 'b': 2, 'c': 34, 'd': 24}

dict_len = range(len(di)-2, len(di))

for l in dict_len:
    d_values = list(itertools.combinations(di.values(), l))
    for d in d_values:
        print d

输出：

如何使值的键也打印出来？喜欢：

a + c = 35
a + b = 3
a + d = 25
b + c = 36
c + d = 58
b + d = 26
a + b + c =37
a + c + d =59
a + b + d =27
c + b + d =60

谢谢。

Answer 1

您可以这样：

di = {'a': 1, 'b': 2, 'c': 34, 'd': 24}

dict_len = range(len(di)-2, len(di))

for l in dict_len:
    d_values = list(itertools.combinations(di.values(), l))
    d_keys =  list(itertools.combinations(di.keys(), l))
    for d,k in zip(d_values,d_keys):
        print(k,sum(d))

哪个生成输出：

('a', 'b') 3
('a', 'c') 35
('a', 'd') 25
('b', 'c') 36
('b', 'd') 26
('c', 'd') 58
('a', 'b', 'c') 37
('a', 'b', 'd') 27
('a', 'c', 'd') 59
('b', 'c', 'd') 60

Answer 2

要获得所需的输出（在问题中指定）：

import itertools

di = {'a': 1, 'b': 2, 'c': 34, 'd': 24}

dict_len = range(len(di)-2, len(di))

for l in dict_len:
    d_values = list(itertools.combinations(di.values(), l))
    d_keys = list(itertools.combinations(di.keys(), l))
    for i in range(len(d_values)):
        print " + ".join(d_keys[i]), "=", sum(d_values[i])

输出：

a + c = 35
a + b = 3
a + d = 25
c + b = 36
c + d = 58
b + d = 26
a + c + b = 37
a + c + d = 59
a + b + d = 27
c + b + d = 60

Answer 3

有趣的问题。您可以这样做：

import itertools

di = {'a': 1, 'b': 2, 'c': 34, 'd': 24}

for n in range(2, len(di)):
  for pairs in itertools.combinations(di.items(), n):
      keys, values = zip(*pairs)  # Note 1
      print("{} = {}".format(' + '.join(keys), sum(values)))

这将导致

a + b = 3
a + c = 35
a + d = 25
b + c = 36
b + d = 26
c + d = 58
a + b + c = 37
a + b + d = 27
a + c + d = 59
b + c + d = 60

注意：

请注意zip(*iterable)构造会更改分组顺序。例如，它将您从[('foo', 1), ('bar', 2)]带到[('foo', 'bar'), (1, 2)]，从而有效地将keys和values分组在一起。

Python，在字典中打印出每种可能组合的元素和的键

3 个答案: