Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- import numpy as np
- import itertools as it
- n=6
- li=[]
- for i in it.product([0, 1], repeat = n**2):
- if (np.reshape(np.array(i), (n, n)).sum(axis=1) < 2).all() and (np.reshape(np.array(i), (n, n)).sum(axis=0)< 2).all() :
- if (np.transpose(np.reshape(np.array(i), (n, n))) != np.reshape(np.array(i), (n, n))).any():
- li.append(np.reshape(np.array(i), (n, n)))
Add Comment
Please, Sign In to add comment