genre counts

This commit is contained in:
CC
2026-09-01 23:01:33 +01:00
parent 1f62a5e008
commit d2ff44e5d5

View File

@@ -28,10 +28,16 @@ def fetch_titles_and_genres(engine: Engine) -> pd.DataFrame:
query = """SELECT title, genres FROM audiobookshelf.ao3;""" query = """SELECT title, genres FROM audiobookshelf.ao3;"""
return run_query(engine, query) return run_query(engine, query)
def count_genre_occurrences(data: pd.DataFrame, column: str = "genres") -> pd.Series:
"""Count how many times each genre appears across all titles."""
return data.explode(column)[column].value_counts()
def main(): def main():
engine = get_engine() engine = get_engine()
data = fetch_titles_and_genres(engine) data = fetch_titles_and_genres(engine)
print(data.head()) genre_counts = count_genre_occurrences(data)
print(genre_counts)
# print(data.head())
if __name__ == "__main__": if __name__ == "__main__":