Numpy - группа данных в сумме значений

Question

Скажем, у меня есть массив значений:

a = np.array([1,5,4,2,4,3,1,2,4]) 

и три «сумма» значения:

b = 10 
c = 9 
d = 7 

Есть ли способ, чтобы сгруппировать значения в a в группы множеств, где значения объединяются в равные b, c и d? Например:

b: [5,2,3] 
c: [4,4,1] 
d: [4,2,1] 

b: [5,4,1] 
c: [2,4,3] 
d: [4,2,1] 

b: [4,2,4] 
c: [5,4] 
d: [1,1,2,3] 

Примечание сумма b, c и d должен оставаться неизменным (== 26). Возможно, эта операция уже имеет имя?

Answer 1

Вот наивная реализация с использованием itertools

from itertools import chain, combinations 

def group(n, iterable): 
    s = list(iterable) 
    return [c for c in chain.from_iterable(combinations(s, r) 
              for r in range(len(s)+1)) 
      if sum(c) == n] 

---

group(5, range(5)) 

Урожайность

[(1, 4), (2, 3), (0, 1, 4), (0, 2, 3)] 

Примечание, это, вероятно, будет очень медленным для больших списков, потому что мы по сути построения и фильтрации через набор мощности этого списка.

---

Вы можете использовать это для

sum_vals = [10, 9, 7] 
a = [1, 5, 4, 2, 4, 3, 1, 2, 4] 

map(lambda x: group(x, a), sum_vals) 

, а затем zip их вместе.