Как совместить тепловую карту и контурный график с Plotly Python - PullRequest
Новая
       24

Как совместить тепловую карту и контурный график с Plotly Python

1 голос
/ 29 февраля 2020

У меня есть два разных графика, одна тепловая карта и контурный график, и я хотел бы поместить контур поверх тепловой карты. Значение z для каждого из графиков соответствует разным величинам. 

import plotly.graph_objects as go

fig1=go.Figure(go.Heatmap(
    x=data.X,
    y=data.Y,
    z=data0.Z1, zsmooth='best',
    colorscale='Viridis'
), layout=layout)

fig2=go.Figure(go.Contour(
    x=data.X,
    y=data.Y,
    z=round(data.Z2,3),
    zmin=0,
    zmax=0.3,
    contours_coloring='lines',
    line_width=2,
    colorscale=[[0,'blue'],[0.5,'black'],[1,'red']],
    contours=dict(
        start=0, end=0.01, size=1)
), layout=layout)

В Альтаире я мог бы просто использовать fig1 + fig2, но это не работает с Plotly. К сожалению, у Альтаира нет контурных возможностей. Если это невозможно с помощью Plotly, можно ли этого достичь с помощью matplotlib?

Редактировать: вот пример данных 

ls = [[-0.1,-3.0,-0.5,0.0],[-0.1,-2.0,-0.25,0.2],[-0.1,-1.0,-0.15,0.25],[-0.1,0.0,0,0.3],[-0.1,1.0,0.1,0.25],[-0.1,2.0,0.25,0.2],[-0.1,3.0,0.5,0.0],[0.0,-3.0,-0.5,0.0],[0.0,-2.0,-0.25,0.1],[0.0,-1.0,-0.15,0.11],[0.0,0.0,0.0,0.1],[0.0,1.0,0.1,0.0],[0.0,2.0,0.25,0.0],[0.0,3.0,0.5,0.0],[0.1,-3.0,-0.5,0.0],[0.1,-2.0,-0.5,0.12],[0.1,-1.0,-0.15,0.12],[0.1,0.0,0.0,0.13],[0.1,1.0,0.1,0.12],[0.1,2.0,0.5,0.12],[0.1,3.0,0.5,0.0]]

data = pd.DataFrame(ls, columns=['X', 'Y', 'Z1', 'Z2'])
...