输入plotlyexpress输出的提示？

Question

如果我有一个函数返回一个绘图表达的图形，类型提示的正确方法是什么？

该图的类型为：

<class 'plotly.graph_objs._figure.Figure'>.

我应该通过单独导入plotly来输入提示，然后编写：

-> plotly.graph_objs._figure.Figure 

？

import pandas as pd
import plotly.express as px
import plotly 
 
long_df = px.data.medals_long()

fig = px.bar(long_df, x="nation", y="count", color="medal", title="Long-Form Input")
print (type (fig))

def plot_bar (df: pd.DataFrame) -> plotly.graph_objs._figure.Figure :
  fig = px.bar(long_df, x="nation", y="count", color="medal", title="Long-Form Input")
  return fig

plot_bar(long_df)

Answer 1

这是一个惯例问题，但通常当人们使用plotly

graph_objects

时，他们有以下含义：

import plotly.graph_objects as go

这意味着您可以将

plotly.graph_objs._figure.Figure

替换为

go.Figure

，这样您的类型提示就更容易阅读：

def plot_bar(df: pd.DataFrame) -> go.Figure():
  fig = px.bar(long_df, x="nation", y="count", color="medal", title="Long-Form Input")
  return fig