pythonplotly：不固定数量的痕迹

3回答

暮色呼如

我认为有一种更灵活的方法可以使用它，plotly.express特别是如果您不想定义颜色。这个想法是正确地转换数据。数据import pandas as pddf = pd.DataFrame({1991:[10,100,50], 1992:[14,30,22], 1993:[16,70,30], 1994:[18,85,65], 1995:[20,30,70], 1996:[42,42,66], 1997:[64,64,60]})df.index = ['US','JAPAN','CN']df = df.T.unstack()\ .reset_index()\ .rename(columns={"level_0": "country", "level_1": "year", 0: "n"})print(df) country year n0 US 1991 101 US 1992 142 US 1993 163 US 1994 184 US 1995 205 US 1996 426 US 1997 647 JAPAN 1991 1008 JAPAN 1992 309 JAPAN 1993 7010 JAPAN 1994 8511 JAPAN 1995 3012 JAPAN 1996 4213 JAPAN 1997 6414 CN 1991 5015 CN 1992 2216 CN 1993 3017 CN 1994 6518 CN 1995 7019 CN 1996 6620 CN 1997 60使用plotly.express现在您的数据是长格式，您可以使用plotly.express如下import plotly.express as pxfig = px.scatter(df, x="year", y="country", size="n", color="country", text="n", size_max=50 # you need this otherwise the bubble are too small )fig.update_layout(plot_bgcolor='rgb(10, 10, 10)', paper_bgcolor='rgb(20, 55, 100)', font={'size': 15, 'family': 'sans-serif', 'color': 'rgb(255, 255, 255)' }, width=1000, height=500, xaxis=dict(title='Output of grapes per year in selected countries', ), showlegend=False, margin=dict(l=100, r=100, t=100, b=100), hovermode = False,)# Uncomment this if you don't wont country as yaxis title# fig.layout.yaxis.title.text = Nonefig.show()

0 0

慕妹3242003

我应该指出，如果您将原始数据附加为文本或可以更轻松地复制和粘贴的内容，那么您的代码将更具可重复性。但是，无论如何，我仍然可以回答您的问题并为您指出正确的方向。您应该做的是使用循环，并从查看行开始data = [trace0, trace1, trace2]。正如您所注意到的，如果您有 100 个国家/地区而不是 3 个国家/地区，此方法将无法扩展。相反，您可以data使用列表理解将其创建为列表，并更新每个跟踪中发生更改的部分。trace0除了国家/地区、值和颜色之外，、trace1、trace2没有太大区别。为了向您展示我的意思，我使用 DataFrame 重新创建了您的数据，然后创建了包含您的国家/地区和颜色的单独列表。# Version 2 could read data from .xlsx file.import plotly as pyimport plotly.graph_objs as goimport openpyxl# wb = openpyxl.load_workbook(('grape output.xlsx'))     # sheet = wb['Sheet1']       # row_max = sheet.max_row# col_max = sheet.max_column# l=[]# for row_n in range(row_max-1):#     l.append([])#     for col_n in range(col_max-1):#         l[row_n].append(sheet.cell(row=row_n+2, column=col_n+2).value)import pandas as pddf = pd.DataFrame({1991:[10,100,50], 1992:[14,30,22], 1993:[16,70,30], 1994:[18,85,65], 1995:[20,30,70], 1996:[42,42,66], 1997:[64,64,60]})df.index = ['US','JAPAN','CN']colors = ['rgb(150,204,90)','rgb(255, 130, 71)','rgb(255, 193, 37)']data = [go.Scatter(    x=df.columns,    y=[country]*len(df.columns),    mode='markers+text',    marker=dict(        color=colors[num],        size= df.loc[country],        showscale = False,        ),    text=list(map(str, df.loc[country])),         textposition='middle center',       )    for num, country in enumerate(df.index)]layout = go.Layout(plot_bgcolor='rgb(10, 10, 10)',                     paper_bgcolor='rgb(20, 55, 100)',                     font={                                      'size': 15,                       'family': 'sans-serif',                       'color': 'rgb(255, 255, 255)'                     },                   width=1000,                   height=500,                   xaxis=dict(title='Output of grapes per year in US, JAPAN and CN', ),                     showlegend=False,                   margin=dict(l=100, r=100, t=100, b=100),                   hovermode = False,                          )# data = [trace0, trace1, trace2]fig = go.Figure(data=data, layout=layout)fig.show()# py.offline.init_notebook_mode()# py.offline.plot(fig, filename='basic-scatter.html')如果我随后向 DataFrame 添加一个包含 1991-1997 年值的测试国家/地区，则不需要更改其余代码，气泡图将相应更新。# I added a test country with datadf = pd.DataFrame({1991:[10,100,50,10], 1992:[14,30,22,20], 1993:[16,70,30,30], 1994:[18,85,65,40], 1995:[20,30,70,50], 1996:[42,42,66,60], 1997:[64,64,60,70]}) df.index = ['US','JAPAN','CN','TEST'] colors = ['rgb(150,204,90)','rgb(255, 130, 71)','rgb(255, 193, 37)','rgb(100, 100, 100)']

0 0

慕田峪9158850

代码已更新到版本 2，可以从 .xlsx 文件读取数据并绘制气泡图。与之前的数据相比，名为“grape output.xlsx”的原始数据添加了新项目：             1991  1992  1993  1994  1995  1996  1997  1998  1999         US    10    14    16    18    20    42    64   100    50      JAPAN   100    30    70    85    30    42    64    98    24         CN    50    22    30    65    70    66    60    45    45      INDIA    90    88    35    50    90    60    40    66    76         UK    40    50    70    50    25    30    22    40    60这是代码：# Version 2 import plotly as pyimport plotly.graph_objs as goimport openpyxlimport pandas as pdwb = openpyxl.load_workbook('grape output.xlsx')sheet = wb['Sheet1']row_max = sheet.max_rowcol_max = sheet.max_columnfirst_row_list = []first_col_list = []for col_n in range(2, col_max+1):    first_row_list.append(sheet.cell(row=1, column=col_n).value)for row_n in range(2,row_max+1):    first_col_list.append(sheet.cell(row=row_n, column=1).value)data_all = pd.read_excel('grape output.xlsx')data = data_all.loc[:,first_row_list]df = pd.DataFrame(data)df.index = first_col_listcolors = ['rgb(150,204,90)','rgb(255, 130, 71)','rgb(255, 193, 37)','rgb(180,240,190)','rgb(255, 10, 1)',          'rgb(25, 19, 3)','rgb(100, 100, 100)','rgb(45,24,200)','rgb(33, 58, 108)','rgb(35, 208, 232)']data = [go.Scatter(    x=df.columns,    y=[country]*len(df.columns),    mode='markers+text',    marker=dict(        color=colors[num],        size= df.loc[country],        showscale = False,        ),    text=list(map(str, df.loc[country])),    textposition='middle center',    )    for num, country in enumerate(reversed(df.index))]layout = go.Layout(plot_bgcolor='rgb(10, 10, 10)',                   paper_bgcolor='rgb(20, 55, 100)',                   font={                       'size': 15,                       'family': 'sans-serif',                       'color': 'rgb(255, 255, 255)'                   },                   width=1000,                   height=800,                   xaxis=dict(title='Output of grapes per year in US, JAPAN and CN'),                   showlegend=False,                   margin=dict(l=100, r=100, t=100, b=100),                   hovermode = False,                   )fig = go.Figure(data=data, layout=layout)py.offline.plot(fig, filename='basic-scatter.html')现在结果是这样的：还有一些小问题：如何去掉 1990 和 2000 这两个数字以及 1990 和 2000 的白色垂直线？如何为1991、1993、1995、1997、1999画白线并以横坐标轴显示所有这些年份？请更正代码 Versinon 2 以改进它。谢谢你！

0 0