refactor dispersion_tutorial.ipynb

This commit is contained in:
2025-03-20 11:15:48 -05:00
parent fdcf31948f
commit 7b1ba996ae
3 changed files with 263 additions and 229 deletions

View File

@@ -1,14 +0,0 @@
import numpy as np
def random_property_rotation(
user_keypad: np.ndarray,
prop_rotation: list[int]
) -> np.ndarray:
transposed = user_keypad.T
if len(prop_rotation) != len(transposed):
raise ValueError("prop_rotation must be the same length as the number of properties")
for idx, prop_set in enumerate(transposed):
rotation = prop_rotation[idx]
rotation = rotation % len(prop_set) if len(prop_set) > 0 else 0
transposed[idx] = np.roll(prop_set, rotation)
return transposed.T