python - 以所有可能的组合拆分数组(不是常规拆分)

Question

假设我有一个数组，

>>> import numpy as np
>>> array  = np.linspace(1,4,4, dtype=np.int)
>>> array
array([1, 2, 3, 4])

我想要一个函数将这个数组拆分成所有可能的部分，这样，

没有拆分:

([1,2,3,4])

分成 2 部分:

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

分成 3 部分:

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

分成 len(array) 部分:

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

我知道有 np.split(array, r)，但它不会给出所有可能的拆分。例如np.split(array, 2) 会给出，

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

如您所见，这不是我需要的。如何实现我的需求？

Answer 1

你可以使用 itertools.combinations根据拆分次数生成在循环内拆分的索引:

>>> from itertools import combinations
>>> [np.split(array, idx) 
...  for n_splits in range(5) 
...  for idx in combinations(range(1, len(array)), n_splits)]
[[array([1, 2, 3, 4])],
 [array([1]), array([2, 3, 4])],
 [array([1, 2]), array([3, 4])],
 [array([1, 2, 3]), array([4])],
 [array([1]), array([2]), array([3, 4])],
 [array([1]), array([2, 3]), array([4])],
 [array([1, 2]), array([3]), array([4])],
 [array([1]), array([2]), array([3]), array([4])]]