From 4df66632718e88c1fe98cdf2ca9f2cc6b5fab9e7 Mon Sep 17 00:00:00 2001
From: David Rotermund <54365609+davrot@users.noreply.github.com>
Date: Mon, 15 May 2023 12:57:07 +0200
Subject: [PATCH] Add files via upload

---
 content/make_drop_map.py | 30 ++++++++++++++++++++++++++++++
 1 file changed, 30 insertions(+)
 create mode 100644 content/make_drop_map.py

diff --git a/content/make_drop_map.py b/content/make_drop_map.py
new file mode 100644
index 0000000..646fb65
--- /dev/null
+++ b/content/make_drop_map.py
@@ -0,0 +1,30 @@
+import numpy as np
+
+
+def make_drop_map(center_x: list[int], center_y: list[int]) -> np.ndarray:
+    center_x_np = np.array(center_x)
+    center_y_np = np.array(center_y)
+
+    center_x_np_unique = np.unique(center_x_np)
+    center_y_np_unique = np.unique(center_y_np)
+
+    organization_matrix = np.full(
+        (center_y_np_unique.shape[0], center_x_np_unique.shape[0]), -1, dtype=np.int64
+    )
+
+    for y in range(0, organization_matrix.shape[0]):
+        to_find_y = center_y_np_unique[y]
+        idx_y = np.where(center_y_np == to_find_y)[0]
+
+        to_find_x = center_x_np[idx_y]
+
+        for x in range(0, to_find_x.shape[0]):
+            idx_x = np.where(center_x_np[idx_y] == to_find_x[x])[0]
+            assert idx_x.shape[0] == 1
+
+            pos_x = np.where(center_x_np_unique == to_find_x[x])[0]
+            assert pos_x.shape[0] == 1
+
+            organization_matrix[y, pos_x] = idx_y[idx_x]
+
+    return organization_matrix