pythonpandasmatplotlibseabornrelplot

How to plot wide format dataframe with seaborn.relplot


I am trying to plot the following line graph with dummy data on 5 cities(C1-C5).

Data frame which has been imported already

Based on what I understand, x="Year", y="Number of Employees" and hue="City". How would I set up the code for it? I have tried doing it in the following manner, but it doesn't work!

Current Code

import seaborn as sns
import pandas as pd

Areas = r'C:\Users\Tachi\Desktop\City.xlsx'
df = pd.read_excel(Areas)
df.set_index('City', inplace=True)

sns.relplot(x="Year", y="Number of Employees",hue="City", kind="line", data=df)

Sample Data

data = {'City': ['C1', 'C2', 'C3', 'C4', 'C5'], 
        2015: [28564, 2585, 4679, 33227, 2000], 
        2016: [83659, 4429, 35834, 1447, 3454], 
        2017: [0, 453, 40903, 46826, 646], 
        2018: [39470, 8364, 29464, 36443, 8364]}
df = pd.DataFrame(data)
df.set_index('City', inplace=True)

       2015   2016   2017   2018
City                            
C1    28564  83659      0  39470
C2     2585   4429    453   8364
C3     4679  35834  40903  29464
C4    33227   1447  46826  36443
C5     2000   3454    646   8364

Solution

  • # transpose the dataframe
    df = df.T
    
    # display(df)
    City     C1    C2     C3     C4    C5
    2015  28564  2585   4679  33227  2000
    2016  83659  4429  35834   1447  3454
    2017      0   453  40903  46826   646
    2018  39470  8364  29464  36443  8364
    
    # plot the dataframe
    sns.relplot(data=df, kind='line', marker='o')
    

    enter image description here

    # set the index of years to a str dtype
    df.index = df.index.astype(str)
    
    # plot the dataframe
    sns.relplot(data=df, kind='line', marker='o')
    

    enter image description here