"""
Subplot
=======
"""

# %%
# The periodic table can be attached as a subplot by specifying the `row` and
# `col` arguments.

import plotly
import ptvis


elements = list(ptvis.Element) * 2
radii = [1]*len(ptvis.Element) + [2]*len(ptvis.Element)

fig = plotly.graph_objects.Figure()
fig.set_subplots(rows=2, cols=1)
fig.add_scatter(x=[0], y=[0], row=1, col=1)  # for ease of viewing
ptvis.attach_polar_bar_cells(fig, elements, radii, row=2, col=1)

plotly.io.show(fig)