Generate dataframe

scverse · Jul 4, 2024 · b4d8dfd · b4d8dfd
1 parent 311f896
commit b4d8dfd
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/inst/scripts/generate_dataframe.py b/inst/scripts/generate_dataframe.py
@@ -0,0 +1,9 @@
+import pandas as pd
+from generate_vector import vector_generators
+
+def generate_dataframe(n_rows, types = None):
+    if types is None:
+        types = list(vector_generators.keys())
+
+    data = {t: vector_generators[t](n_rows) for t in types}
+    return pd.DataFrame(data)