Browse Source

Fix missing x-axis tick in weekly forecast component plot (#28)

* prepend None value to list of days in weekly forecast plot

* set the x ticks with list of days in weekly forecast component plot
Christian Stade-Schuldt 8 years ago
parent
commit
143e544beb
1 changed files with 1 additions and 0 deletions
  1. 1 0
      python/fbprophet/forecaster.py

+ 1 - 0
python/fbprophet/forecaster.py

@@ -721,6 +721,7 @@ class Prophet(object):
                 ax.fill_between(range(len(days)), y_weekly_l, y_weekly_u,
                 ax.fill_between(range(len(days)), y_weekly_l, y_weekly_u,
                                 color=forecast_color, alpha=0.2)
                                 color=forecast_color, alpha=0.2)
             ax.grid(True, which='major', c='gray', ls='-', lw=1, alpha=0.2)
             ax.grid(True, which='major', c='gray', ls='-', lw=1, alpha=0.2)
+            ax.set_xticks(range(len(days)))
             ax.set_xticklabels(days)
             ax.set_xticklabels(days)
             ax.set_xlabel('Day of week')
             ax.set_xlabel('Day of week')
             ax.set_ylabel('weekly')
             ax.set_ylabel('weekly')