Python: Как сделать заштрихованные области или чередующийся цвет фона, используя плотно?

Question

Использование только этих нескольких строк кода из plot.ly даст вам график ниже в блокноте jupyter:

Фрагмент 1:

import plotly
import cufflinks as cf
from plotly.offline import download_plotlyjs, init_notebook_mode, plot, iplot
init_notebook_mode(connected=True)

iplot(cf.datagen.lines().iplot(asFigure=True,
                               kind='scatter',xTitle='Dates',yTitle='Returns',title='Returns'))

Сюжет 1:

Как вы можете настроить его так, чтобы вы могли иметь чередующиеся цвета bakcground вграфик ниже, как это было показано в этом посте с использованием matplotlib ?

Вот ссылка , которая объясняет, как добавить затененные области, например так:

Фрагмент 2:

df.iplot(vspan={'x0':'2015-02-15','x1':'2015-03-15','color':'rgba(30,30,30,0.3)','fill':True,'opacity':.4}, 
         filename='cufflinks/custom-regions')

График 2:

Спасибо за любые предложения!

Answer 1

Как предлагается в вопросе, возможное решение может лежать в функции vspan. Тем не менее, казалось, что гораздо проще добавить несколько затененных областей для оси Y с помощью hspan, чем в случае с vspan и осью X. Последнему нужно немного доработать. Более подробную информацию можно найти после моего предложенного решения.

---

Следующий фрагмент создается с помощью фрагмента и функции multiShades ниже:

Участок:

Отрывок:

### Setup from the question ###

import plotly
import cufflinks as cf
from plotly.offline import download_plotlyjs, init_notebook_mode, plot, iplot
import pandas as pd
import numpy as np
from IPython.display import HTML
from IPython.core.display import display, HTML
import copy

# setup
init_notebook_mode(connected=True)
np.random.seed(123)
cf.set_config_file(theme='pearl')

# Random data using cufflinks
df = cf.datagen.lines()

fig = df.iplot(asFigure=True, kind='scatter',
               xTitle='Dates',yTitle='Returns',title='Returns',
               vspan={'x0':'2015-01-11','x1':'2015-02-22','color':'rgba(30,30,30,0.3)','fill':True,'opacity':.4})

### ANSWER ###

xStart = ['2015-01-11', '2015-02-08', '2015-03-08', '2015-04-05']
xStop = ['2015-01-25', '2015-02-22', '2015-03-22', '2015-04-10']

def multiShades(plot, x0, x1):
    """ Adds shaded areas for specified dates in a plotly plot.
        The lines of the areas are set to transparent using rgba(0,0,0,0)
    """
    # get start and end dates
    x0 = xStart
    x1 = xStop

    # get dict from tuple made by vspan()
    xElem = fig['layout']['shapes'][0]

    # container (list) for dicts / shapes
    shp_lst=[]

    # make dicts according to x0 and X1
    # and edit elements of those dicts
    for i in range(0,len(x0)):
        shp_lst.append(copy.deepcopy(xElem))
        shp_lst[i]['x0'] = x0[i]
        shp_lst[i]['x1'] = x1[i]
        shp_lst[i]['line']['color'] = 'rgba(0,0,0,0)'

    # replace shape in fig with multiple new shapes
    fig['layout']['shapes']= tuple(shp_lst)
    return(fig)

fig = multiShades(plot=fig, x0=xStart, x1=xStop)

iplot(fig)

Некоторые детали:

Функция vspan 'заполняет' кортеж fig['layout']['shapes'] словарем вида:

{'fillcolor': 'rgba(187, 187, 187, 0.4)',
 'line': {'color': '#BBBBBB', 'dash': 'solid', 'width': 1},
 'type': 'rect',
 'x0': '2015-01-11',
 'x1': '2015-02-22',
 'xref': 'x',
 'y0': 0,
 'y1': 1,
 'yref': 'paper'}

Моя функция просто берет этот словарь, делает несколько копий, редактирует эти копии в соответствии с аргументами функции и заменяет исходный кортеж новым кортежем из функции.

Задача:

Этот подход может быть немного сложнее, когда добавлено больше фигур. Кроме того, даты должны быть жестко запрограммированы - по крайней мере, пока кто-нибудь не найдет ответ на Как получить значения для основных тиков и линий сетки?