一维数组的排列组合，可以通过矩阵运算来实现吗？

Question

对于一列数组的排列组合，可以通过双层循环来实现。

那么对于 NumPy 和 MATLAB 这些高级数据科学语言来说，有没有更简便的方法或语句来代替双层循环呢？

可以是数学上矩阵知识的解决方法，也可以是语言库中封装的函数。

Answer 1

排列：itertools.permutations(iterable, r=None)
组合：itertools.combinations(iterable, r)

例子：

>>> from itertools import permutations, combinations
>>> a = [1, 2, 3]
>>> list(permutations(a))
[(1, 2, 3), (1, 3, 2), (2, 1, 3), (2, 3, 1), (3, 1, 2), (3, 2, 1)]
>>> for r in range(len(a)):
...     print(list(combinations(a, r+1)))
[(1,), (2,), (3,)]
[(1, 2), (1, 3), (2, 3)]
[(1, 2, 3)]