Best cuisines: insight 1

bestcuisines_insight_1.py

fig = plt.figure()
for index, city in enumerate(cities):
    # For each city, get the top 10 cuisines and plot them in a horizontal bar graph
    ax = fig.add_subplot(3, 3, index+1)

    # Get the top 10 cuisines for the city
    cuisines = cuisines_city_normalized.loc[city]
    cuisines = cuisines.sort_values(ascending=False).head(10).sort_values(ascending=True)
    
    # Plot in red the bar with the local cuisine, plot in blue the rest
    local_cuisine = city_local_cuisines[city]
    color = ''
    for i in cuisines.keys():
        if i == local_cuisine:
            color = color + 'r'
        else:
            color = color + 'b'
    
    # Plot the top 10 cuisines
    cuisines.plot(kind='barh', ax=ax, figsize=(16, 14), color=list(color))
    ax.set_title(city)
    ax.set_xlabel("% of restaurants")