import plotly.io as pio
import plotly.express as px
import plotly.graph_objects as go
from plotly.subplots import make_subplots
import pandas as pd
import numpy as np

# 設(shè)置plotly默認(rèn)主題
pio.templates.default = 'plotly_white'

# 設(shè)置pandas打印時(shí)顯示所有列
pd.set_option('display.max_columns', None)

df = px.data.gapminder().query("year==2007")
fig = px.scatter(df, x="gdpPercap", y="lifeExp", color="continent",
? ? size="pop", size_max=45, log_x=True)

fig.update_layout(legend=dict(
? ? yanchor="top",
? ? y=0.99,
? ? xanchor="left",
? ? x=0.01
))

fig.write_image('../pic/legend_1.png', scale=2)
fig.show()

df = px.data.gapminder().query("year==2007")
fig = px.scatter(df, x="gdpPercap", y="lifeExp", color="continent",
? ? size="pop", size_max=45, log_x=True)

fig.update_layout(legend=dict(
? ? orientation="h",
? ? yanchor="bottom",
? ? y=1.02,
? ? xanchor="center",
? ? x=0.5,
? ? title_text=''
))

fig.write_image('../pic/legend_2.png', scale=2)
fig.show()

df = px.data.gapminder().query("year==2007")
fig = px.scatter(df, x="gdpPercap", y="lifeExp", color="continent",
? ? size="pop", size_max=45, log_x=True)


fig.update_layout(
? ? legend=dict(
? ? ? ? x=0,
? ? ? ? y=1,
? ? ? ? traceorder="reversed",
? ? ? ? title_font_family="Times New Roman",
? ? ? ? font=dict(
? ? ? ? ? ? family="Courier",
? ? ? ? ? ? size=12,
? ? ? ? ? ? color="black"
? ? ? ? ),
? ? ? ? bgcolor="LightSteelBlue",
? ? ? ? bordercolor="Black",
? ? ? ? borderwidth=2
? ? )
)

fig.write_image('../pic/legend_3.png', scale=2)
fig.show()

fig = go.Figure()

# 使用 name 參數(shù)指定條目文本，legendrank 指定順序
fig.add_trace(go.Bar(name="fourth", x=["a", "b"], y=[2,1], legendrank=4))
fig.add_trace(go.Bar(name="second", x=["a", "b"], y=[2,1], legendrank=2))
fig.add_trace(go.Bar(name="first", x=["a", "b"], y=[1,2], legendrank=1))
fig.add_trace(go.Bar(name="third", x=["a", "b"], y=[1,2], legendrank=3))

fig.write_image('../pic/legend_4.png', scale=2)
fig.show()

fig = go.Figure()

fig.add_trace(go.Scatter(
? ? x=[1, 2, 3],
? ? y=[2, 1, 3],
? ? legendgroup="group", ?# this can be any string, not just "group"
? ? legendgrouptitle_text="First Group Title",
? ? name="first legend group",
? ? mode="markers",
? ? marker=dict(color="Crimson", size=10)
))

fig.add_trace(go.Scatter(
? ? x=[1, 2, 3],
? ? y=[2, 2, 2],
? ? legendgroup="group",
? ? name="first legend group - average",
? ? mode="lines",
? ? line=dict(color="Crimson")
))

fig.add_trace(go.Scatter(
? ? x=[1, 2, 3],
? ? y=[4, 9, 2],
? ? legendgroup="group2",
? ? legendgrouptitle_text="Second Group Title",
? ? name="second legend group",
? ? mode="markers",
? ? marker=dict(color="MediumPurple", size=10)
))

fig.add_trace(go.Scatter(
? ? x=[1, 2, 3],
? ? y=[5, 5, 5],
? ? legendgroup="group2",
? ? name="second legend group - average",
? ? mode="lines",
? ? line=dict(color="MediumPurple")
))

fig.update_layout(title="Try Clicking on the Legend Items!")

fig.write_image('../pic/legend_5.png', scale=2)
fig.show()

fig = go.Figure()

fig.add_trace(go.Scatter(
? ? x=[1, 2, 3, 4, 5],
? ? y=[1, 2, 3, 4, 5],
))

fig.add_trace(go.Scatter(
? ? x=[1, 2, 3, 4, 5],
? ? y=[5, 4, 3, 2, 1],
? ? visible='legendonly'
))

fig.write_image('../pic/legend_6.png', scale=2)
fig.show()

fig = go.Figure()

fig.add_trace(go.Scatter(
? ? x=[1, 2, 3, 4, 5],
? ? y=[1, 2, 3, 4, 5],
? ? showlegend=False
))


fig.add_trace(go.Scatter(
? ? x=[1, 2, 3, 4, 5],
? ? y=[5, 4, 3, 2, 1],
))

fig.update_layout(showlegend=True)

fig.write_image('../pic/legend_7.png', scale=2)
fig.show()

fig = go.Figure()

fig.add_trace(go.Scatter(
? ? x=[1, 2, 3, 4, 5],
? ? y=[1, 2, 3, 4, 5],
? ? mode='markers',
? ? marker={'size':10}
))

fig.add_trace(go.Scatter(
? ? x=[1, 2, 3, 4, 5],
? ? y=[5, 4, 3, 2, 1],
? ? mode='markers',
? ? marker={'size':100}
))

fig.update_layout(legend= {'itemsizing': 'trace'})

fig.write_image('../pic/legend_8.png', scale=2)
fig.show()