Matplotlib 2 y axis with mean

Question

Hi i have been trying to figure out how do i plot my data frame into the graphs. My dataframe looks like this .

Country | exports 2015 | exports 2016 | Gdp 2015 | GDP 2016| 
  A     |     500      |     600      |   34324  |  23525  | 
  B     |     435      |     335      |    3243  |   2324  |
  C     |     222      |     324      |    23423 |   1233  | 
  D     |     7756     |     9000     |    32424 |  65545  |

Basically i want to compare the mean for all the columns and plot them on a graph with 1 x axis as the years and 2 y axis as the exports and GDP. I can only do for 1 year. Basically im trying to get

         |                                    |
         |                                    |
         |                                    |
 Mean    |                                    |
Exports  |                                    | Mean GDP
         |                                    |
         |                                    |
         |____________________________________|  
              2015                  2016

Do i need to somehow convert the data into a mean when plotting the graph or do i do another column? Any advice would be great thanks :)

You are basically asking how to use two y axis in the plot. [Here](https://stackoverflow.com/questions/14762181/adding-a-y-axis-label-to-secondary-y-axis-in-matplotlib), [here](https://stackoverflow.com/questions/45155510/plotting-mulitple-lines-on-two-y-axis-using-matplotlib), [here](https://matplotlib.org/gallery/api/two_scales.html), are some of the many links that will solve your problem. You just need to search for computing mean in DataFrames. I can provide their links too but I guess you got the point. I am voting to close this question as a duplicate. — Sheldore, Sep 11 '18 at 10:01
[Here](https://stackoverflow.com/questions/15082682/matplotlib-diagrams-with-2-y-axis) and [here](https://stackoverflow.com/questions/36104251/matplotlib-pyplot-how-to-set-up-a-second-y-axis-for-an-existing-plot) are two more links for you — Sheldore, Sep 11 '18 at 10:01
This looks like two questions, one a `pandas` question and one a `matplotlib` question. Might be better to split it up. Will make searching for answers (use google with `site:stackoverflow.com`) easier too. — Daniel F, Sep 11 '18 at 10:02
[Here](https://stackoverflow.com/questions/31037298/pandas-get-column-average-mean), [here](https://stackoverflow.com/questions/40739558/pandas-compute-the-mean-of-a-column-grouped-by-another-column) and [here](https://stackoverflow.com/questions/16689514/how-to-get-the-average-of-dataframe-column-values) are some links on how to compute the mean in pandas — Sheldore, Sep 11 '18 at 10:05
Possible duplicate of [Adding a y-axis label to secondary y-axis in matplotlib](https://stackoverflow.com/questions/14762181/adding-a-y-axis-label-to-secondary-y-axis-in-matplotlib) — Alexander McFarlane, Sep 11 '18 at 10:26

An economist · Answer 1 · 2018-09-11T13:29:27.043

This is one possible solution, using pandas. The only difficulty is with setting the legend position, since you have to set labels for each y-axis. Keep in mind that double axis plots are very confusing.

import pandas as pd
import matplotlib.pyplot as plt


# Stacked input data 
df = pd.DataFrame({'Country': ['A','B', 'C', 'D','A','B', 'C', 'D'],
                   'Year': ['2015','2015','2015','2015','2016','2016','2016','2016'],
                   'Export': [500, 435, 222, 7756,600, 335, 324, 9000],
                   'GDP': [34324, 3243, 23423, 32424,23525, 2324, 1233, 65545]})

# Calculate yearly means
year_means = df.groupby('Year').mean().reset_index()

# Plot the means
ax = year_means.plot(x='Year',
                     y=['Export', 'GDP'],
                     secondary_y= 'GDP',
                     kind= 'bar',
                     mark_right=False)

#Set labels
ax.set_ylabel('Exports')
ax.right_ax.set_ylabel('GDP')

# Adjust legend position
ax.legend(bbox_to_anchor=(1,1), loc="upper left")
ax.right_ax.legend(bbox_to_anchor=(1.2,1), loc="upper left")

plt.show()

EDIT: OP does not have stacked input data. One way to fix it is to transform variables individually and then combine them into single frame. Below solution is far from optimal.

# Not stacked input data 
df = pd.DataFrame({'Country': ['A','B', 'C', 'D'],
                   'Export 2015': [500, 435, 222, 7756],
                   'Export 2016': [600, 335, 324, 9000],
                   'GDP 2015': [34324, 3243, 23423, 32424],
                   'GDP 2016': [23525, 2324, 1233, 65545]})


def stack_variable(df, variable):

    # Get columns of the input dataframe
    names = df.columns

    # Get column names with variable of interest
    var_columns = [name for name in names if variable in name]

    # Extract years
    years = [y.split(variable + ' ')[1] for y in var_columns]

    # Empty dataframe to store results
    stacked_df = pd.DataFrame(columns = [variable, 'Year'])

    # Fill the empty frame
    for idx, col in enumerate(var_columns):
             current = pd.DataFrame({variable: df[col],
                                     'Year': years[idx]})

             stacked_df = stacked_df.append(current)



    return stacked_df


exports = stack_variable(df, 'Export')
gdp = stack_variable(df, 'GDP')

stacked_df = pd.concat([exports, gdp['GDP']], axis=1).reset_index(drop=True)

Which would return:

stacked_df

       Export   Year    GDP
    0   500     2015    34324
    1   435     2015    3243
    2   222     2015    23423
    3   7756    2015    32424
    4   600     2016    23525
    5   335     2016    2324
    6   324     2016    1233
    7   9000    2016    65545

i get what you mean. However , my dataframe is exactly like the one shown in my question above. i cant get it to stack properly because they have different columns names? For e.g GDP 2015 and GDP 2016. This is because im also reading it from a csv file — user10346242, Sep 11 '18 at 12:09
Show some initiative, you can probably adjust the csv file or write simple function to do that... See my edit. — An economist, Sep 11 '18 at 13:24

Matplotlib 2 y axis with mean

1 Answers1