Создать полную комбинацию без циклических вращений ⇐ Python

[Anonymous]

Мне нужно сгенерировать двоичную последовательность определенной длины.
Для генерации комбинаций я использую библиотеку numpy. Я нашел следующий код:

Код: Выделить всё

np.array(np.meshgrid(*[[1, 0] for x in range(size)])).T.reshape(-1, size)

Он работает, и работает быстро, но создает все комбинации определенной длины.
if size = 3

Код: Выделить всё

[1 1 1]
[1 0 1]
[0 1 1]
[0 0 1]
[1 1 0]
[1 0 0]
[0 1 0]
[0 0 0]

Мне нужно сгенерировать комбинации таким образом, чтобы удалить все циклические копии, т.е.:

[1 1 1]
[1 0 1]
[0 1 1] потому что [1 0 1] и [1 1 0] — циклические смещения
[0 0 1]
[1 1 0]
[1 0 0]
[0 1 0]
[0 0 0]

Мне нужно написать функцию, которая будет генерировать все разрешенные комбинации любой длины от 3 до 1000 элементов.
Кажется, это работает:

Код: Выделить всё

def gen_matrix(size):
return np.array(np.meshgrid(*[[1, 0] for x in range(0, size)])).T.reshape(-1, size)

def build_matrix(size_row):
a = gen_matrix(size_row)
kill = True
while kill:
kill = False
for i in a:
if kill:
break
for j in range(1, size_row):
b = np.roll(i, j)
if np.array_equal(i, b):
continue
k = np.where((a == b).all(axis=1))
if np.size(k) > 0:
a = np.delete(a, k[0], 0)
kill = True
return a

Результат:

Код: Выделить всё

[1 1 1 1 1]
[1 0 1 1 1]
[0 0 1 1 1]
[0 1 0 1 1]
[0 0 0 1 1]
[0 0 1 0 1]
[0 0 0 0 1]
[0 0 0 0 0]

Какой способ лучше?


Подробнее здесь: https://stackoverflow.com/questions/730 ... -rotations