Changes to plot. Contributing towards issue - 217

[aniket-ak]

Description

In this PR, there are two changes

1. The plots created for the published dataset containing 7 countries are such that the X axis contains the countries and Y axis contains the months. It should be other way around. Each line in the plot must represent one country each. This is fixed
2. The x value in plot_extrapolated_country is hardcoded to 53. I changed this to take the shape of the incoming numpy array to avoid issues when the shape changes

Affected Dependencies

None

How has this been tested?

• I have tested the changes on my local machine

Checklist

• I have followed the Contribution Guidelines and Code of Conduct
• I have commented my code following the OpenMined Styleguide
• I have labeled this PR with the relevant Type labels
• My changes are covered by tests

[review-notebook-app]

Check out this pull request on 

See visual diffs & provide feedback on Jupyter Notebooks.

---

Powered by ReviewNB

[aniket-ak]

@shubham3121 FYI.