Ich verwende die Numpy-Bibliothek, um Kombinationen zu generieren. Ich habe den folgenden Code gefunden:
Code: Select all
np.array(np.meshgrid(*[[1, 0] for x in range(size)])).T.reshape(-1, size)
if size = 3
Code: Select all
[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], weil [1 0 1] und [1 1 0] zyklische Verschiebungen sind
[0 0 1]
[1 1 0]
[1 0 0]
[0 1 0]
[0 0 0]
Ich muss eine Funktion schreiben, die alle zulässigen Kombinationen beliebiger Länge von 3 bis 1000 Elementen generiert.
Das scheint zu funktionieren:
Code: Select all
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
Code: Select all
[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]
 Mobile version