假设您有一个多维数组
a = np.array([[1, 2], [3, 5], [4,5], [9,5]])我希望将两个数组的所有可能组合都给出多维数组"a“,以便:
[[1, 2],[3, 5]] , [[1, 2],[4,5]] ... 我不知道该怎么做。有人有什么建议和建议吗?
发布于 2022-09-20 16:30:43
您可以使用在itertools.combinations中定义的这个答案函数。此代码创建所有组合的列表。
import numpy as np
import itertools
a = np.array([[1, 2], [3, 5], [4,5], [9,5]])
combination=[]
for L in range(len(a) + 1):
for subset in itertools.combinations(a, L):
combination.append([list(sub) for sub in subset])
combination https://stackoverflow.com/questions/73789717
复制相似问题