Я пытаюсь найти решение здесь, чтобы сформировать N групп из списка чисел, чтобы N групп были сформированы со всеми комбинациями чисел из списка. Например, если у меня есть список чисел с [1,2,3,4,5] и N, равным 2, тогда мой результат должен вернуть все комбинации, образующие 2 группы с заданными числами. Я использую pyspark для решения этой проблемы с массивом чисел в столбце данных. Вот пример результата, который я ожидаю для примера, приведенного в моем вопросе -
[[1][2,3,4,5]], [[2],[1,3,4,5]], [[3],[1,2,4,5]], [[4],[1,2,3,5]], [[5],[1,2,3,4]],
[[1,2],[3,4,5]], [[1,3],[2,4,5]], [[1,4],[2,3,5]], [[1,5],[2,3,4]],
[[2,3],[1,4,5]], [[2,4],[1,3,5]], [[2,5],[1,3,4]],
[[3,4],[1,2,5]], [[3,5],[1,2,4]],
[[4,5],[1,2,3]]