From c8cf83eedf90fae385ae63b9d0b722913015c15e Mon Sep 17 00:00:00 2001
From: Grant Sanderson <grant@3blue1brown.com>
Date: Tue, 15 Aug 2023 20:40:15 -0700
Subject: [PATCH] Add cartesian_product

---
 manimlib/utils/iterables.py | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/manimlib/utils/iterables.py b/manimlib/utils/iterables.py
index 3b7dbfba..90cd4c4b 100644
--- a/manimlib/utils/iterables.py
+++ b/manimlib/utils/iterables.py
@@ -136,6 +136,18 @@ def array_is_constant(arr: np.ndarray) -> bool:
     return len(arr) > 0 and (arr == arr[0]).all()
 
 
+def cartesian_product(*arrays: np.ndarray):
+    """
+    Copied from https://stackoverflow.com/a/11146645
+    """
+    la = len(arrays)
+    dtype = np.result_type(*arrays)
+    arr = np.empty([len(a) for a in arrays] + [la], dtype=dtype)
+    for i, a in enumerate(np.ix_(*arrays)):
+        arr[..., i] = a
+    return arr.reshape(-1, la)
+
+
 def hash_obj(obj: object) -> int:
     if isinstance(obj, dict):
         return hash(tuple(sorted([