Spaces:
Sleeping
Sleeping
File size: 2,681 Bytes
90a6c5f 5a0a701 90a6c5f 5a0a701 4d9b242 50f339f 53d94e8 6da5cf7 90a6c5f 5a0a701 dd67cb4 0488555 514acd5 90a6c5f 5a0a701 90a6c5f 4d9b242 90a6c5f dcbc569 514acd5 dcbc569 4d9b242 dcbc569 4d9b242 dcbc569 a178b93 514acd5 dcbc569 280aeca a178b93 514acd5 0488555 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 |
import streamlit as st
import pandas as pd
import plotly.express as px
# Load your dataframes
df_tv_series= pd.read_csv('series_after_cleaning.csv')
df_movies= pd.read_csv('movie_after_cleaning.csv')
df_movies['genre'] = df_movies['genre'].str.split(',')
df_tv_series['genre'] = df_tv_series['genre'].str.split(',')
df_movies['country'] = df_movies['country'].str.split(',')
df_tv_series['country'] = df_tv_series['country'].str.split(',')
# Function to generate treemap
def create_treemap(df, title):
fig = px.treemap(df, path=['parentalguide'], title=title)
return fig
def create_country_map(df, title):
# Explode the country column to count each country separately
df_exploded = df.explode('country')
country_counts = df_exploded['country'].value_counts().reset_index()
country_counts.columns = ['country', 'count']
fig = px.choropleth(locations=country_counts['country'], locationmode='country names',
color=country_counts['count'], hover_name=country_counts['country'],
title=title, color_continuous_scale=px.colors.sequential.Plasma)
fig.update_layout(showlegend=False)
return fig
def create_country_map(df, title):
country_counts = df['country'].value_counts().reset_index()
country_counts.columns = ['country', 'count']
fig = px.choropleth(country_counts, locations="country", locationmode='country names',
color="count", hover_name="country",
color_continuous_scale=px.colors.sequential.Plasma, title=title)
return fig
# Streamlit app
st.title('Parental Guide Treemaps')
# Split into two columns
col1, col2 = st.columns(2)
# Initialize variable for selection
selection = 'Movies'
# Add buttons in each column
with col1:
if st.button('Movies'):
selection = 'Movies'
with col2:
if st.button('TV Series'):
selection = 'TV Series'
# Display the corresponding treemap in the center
if selection == 'Movies':
st.plotly_chart(create_treemap(df_movies, 'Parental Guide - Movies'), use_container_width=True)
st.plotly_chart(create_genre_bar_chart(df_movies, 'Top 10 Genres - Movies'), use_container_width=True)
st.plotly_chart(create_country_map(df_movies, 'Global Distribution of Movies'), use_container_width=True)
elif selection == 'TV Series':
st.plotly_chart(create_treemap(df_tv_series, 'Parental Guide - TV Series'), use_container_width=True)
st.plotly_chart(create_genre_bar_chart(df_tv_series, 'Top 10 Genres - TV Series'), use_container_width=True)
st.plotly_chart(create_country_map(df_tv_series, 'Global Distribution of TV Series'), use_container_width=True)
|