Plotly: Bagaimana cara menambahkan quivers ke subplot?

Question

Saya ingin menambahkan quivers ke plotlysubplot. Tetapi saya tidak dapat menemukan cara untuk melakukan ini.

Inilah yang ingin saya lakukan:

fig = plotly.subplots.make_subplots(rows=1, cols=2)
    
    
x,y = np.meshgrid(np.arange(0, 2, .2), np.arange(0, 2, .2))
u = np.cos(x)*y
v = np.sin(x)*y
    
quivers = ff.create_quiver(x, y, u, v)
    
fig.add_trace(data=quivers.data,col=1,row=1)

Masalahnya adalah add_tracetidak menerima argumen data, dan add_tracestidak menerima argumen kolom dan baris.

Saat meluncurkan bagian kode ini, saya mendapatkan kesalahan berikut:

TypeError: add_trace() got an unexpected keyword argument 'data'

Terima kasih!

Answer 1

Tampaknya objek gambar lengkap tidak dapat digunakan untuk menyiapkan subplot menggunakan make_subplots(). Dan seperti yang Anda tunjukkan pada diri Anda sendiri, fig.add_trace()tidak dapat digunakan secara langsung dengan data dari ff.create_quiver()gambar. Apa yang akan berhasil, adalah memasukkan unik go.Scatteruntuk setiap dan setiap elemen x dan y di fig1.data:

'x': [0.0, 0.0, None, ..., 1.7591036229552444, 1.7526465527333175, None],
'y': [0.0, 0.0, None, ..., 1.9752925735580753, 1.9216800167812427, None]

Itu mungkin terdengar sedikit rumit, tetapi sebenarnya tidak sama sekali. Buat saja dua ff.create_quiver() figuresdan gunakan ini untuk masing-masing:

# add all fig1.data as individual traces in fig at row=1, col=1
for d in fig1.data:
    subplots.add_trace(go.Scatter(x=d['x'], y=d['y']),
                  row=1, col=1)

Menggunakan potongan di bawah ini akan menghasilkan penyiapan subplot berikut dengan 1 baris dan dua kolom. Bahkan bentuk panah untuk semua garis disertakan.

Merencanakan

Kode lengkap

import numpy as np
from plotly.subplots import make_subplots
import plotly.graph_objects as go
import plotly.figure_factory as ff

# data
x,y = np.meshgrid(np.arange(0, 2, .2), np.arange(0, 2, .2))
u = np.cos(x)*y
v = np.sin(x)*y

# quiver plots
fig1 = ff.create_quiver(x, y, u, v)
fig2 = ff.create_quiver(x, y, u*0.9, v*1.1)
    
# subplot setup
subplots = make_subplots(rows=1, cols=2)

# add all fig1.data as individual traces in fig at row=1, col=1
for d in fig1.data:
    subplots.add_trace(go.Scatter(x=d['x'], y=d['y']),
                  row=1, col=1)

# add all fig2.data as individual traces in fig at row=1, col=2
for d in fig1.data:
    subplots.add_trace(go.Scatter(x=d['x'], y=d['y']),
                  row=1, col=2)
subplots.show()

Answer 2

Saya menemukan solusi di halaman web plotly dengan menyesuaikan sumbu secara manual daripada menggunakan make_subplots.

import numpy as np
import plotly.figure_factory as ff
import plotly.graph_objects as go

x,y = np.meshgrid(np.arange(0, 2, .2), np.arange(0, 2, .2))
u = np.cos(x)*y
v = np.sin(x)*y

fig1 = ff.create_quiver(x, y, u, v, name='trace1')
fig2 = ff.create_quiver(x, y, u*0.9, v*2, name='trace2')


for i in range(len(fig1.data)):
    fig1.data[i].xaxis = 'x1'
    fig1.data[i].yaxis = 'y1'

for i in range(len(fig2.data)):
    fig2.data[i].xaxis = 'x2'
    fig2.data[i].yaxis = 'y2'

fig1.layout.xaxis1.update({'anchor': 'y1', 'domain': [0.55, 1]})
# apparently [0.55, 1] is relative to the full chart's dimensions
fig1.layout.yaxis1.update({'anchor': 'x1'})

fig2['layout']['xaxis2'] = {'anchor': 'y2', 'domain': [0, 0.45]}
fig2['layout']['yaxis2'] = {'anchor': 'x2'}

fig = go.Figure()
fig.add_traces([fig1.data[0], fig2.data[0]])
fig.layout.update(fig1.layout)
fig.layout.update(fig2.layout)