Optimize data sort order

dataforgoodfr · May 14, 2024 · a43eb55 · a43eb55
1 parent e62d5ba
commit a43eb55
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/observable/src/data/films.sqlite.py b/observable/src/data/films.sqlite.py
@@ -70,6 +70,9 @@
 # Set original title to blank string if same as title
 df["original_title"] = df["original_title"].where(df["title"] != df["original_title"], "")
 
+# Sort dataframe by 'popularity' in descending order
+df = df.sort_values(by="popularity", ascending=False)
+
 # Save the dataframe to a SQLite database
 with tempfile.NamedTemporaryFile(suffix=".sqlite", delete=False) as temp_file:
     temp_filename = temp_file.name

diff --git a/observable/src/data/series.sqlite.py b/observable/src/data/series.sqlite.py
@@ -35,6 +35,9 @@
     "",
 )
 
+# Sort dataframe by 'popularity' in descending order
+df = df.sort_values(by="popularity", ascending=False)
+
 # Save the dataframe to a SQLite database
 with tempfile.NamedTemporaryFile(suffix=".sqlite", delete=False) as temp_file:
     temp_filename = temp_file.name