为条形图中的迹线编写函数

问题描述

我创建了一个数据透视表:

I created a pivot table:

df = data.pivot_table(index='col_A', columns='col_B', values='col_C', fill_value=0)

数据帧df是10 * 25数据帧.我想将单个图形中的所有列绘制为迹线.但是，为所有25条轨迹编写代码将非常繁琐.有没有办法我可以写一个像这样的函数:

The dataframe df is a 10*25 dataframe. I want to plot all the columns in a single graph as traces. However, it would be super tedious to write code for all 25 traces. Is there a way I can write a function like:

import plotly.plotly as py
import plotly.tools as tls
from plotly.offline import download_plotlyjs, init_notebook_mode, plot, iplot
import plotly.graph_objs as go
init_notebook_mode(connected=True)
iplot([go.Bars(x=df.index, y=df[col], name=col for col in df.columns)])

上面的代码给我一个语法错误的错误.

The above code is giving me an error of invalid syntax.

推荐答案

为您的更好理解，我创建了一个示例:

I am created example for your better understanding:

# import what we need
import plotly
import plotly.graph_objs as go
import pandas as pd
# Create DataFrame
df = pd.DataFrame({"A":[1,1,1,0],
                   "B":[1,1,0,0],
                   "C":[1,0,0,0]
                   })
# Convert names of columns in a list
traceslist = df.columns.tolist()
# Check list
print(traceslist)
# Create a function that will create as many traces for us as we need
def tracing(column):
   trace = go.Bar(
         x = df.index,
         y = df[column],
         # Parameters above specify what you would see if hover on any column
         name = column,
         text=column,
         textposition='auto',
         hoverinfo="x+y")
   return trace
# Create data
data = []
# Fill out data with our traces
for i in range(len(traceslist)):
   eachtrace = tracing(traceslist[i])
   data.append(eachtrace)
# Optional: create layout
layout = go.Layout(
      # Set title to plot
      title = "Bam!",
      # Choose one of the barmode below and comment another
      barmode="stack",
      #barmode="group"
      )
# Create figure with all we need to plot
fig = go.Figure(data=data, layout=layout)
# Use offline plot without connection to plotly site
plotly.offline.plot(fig, filename='Bam.html')

在上面的代码中，只需使用函数即可创建traces.并在其后使用for循环获取data中的所有traces.您可以在这里看到两个barmode参数之间的区别:1)barmode="stack";2)barmode="group".

In code above just use function to create traces. And using for loop after it to get all the traces in data.What difference between two barmode parameter you can see here:1) barmode="stack";2) barmode="group".

您可以创建更漂亮的图表(请查看条形图的官方文档) ，只需指定其他参数即可.

You can create a much more beautiful plots (check official docs for bar chart), just specify additional parameters.

这篇关于为条形图中的迹线编写函数的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持！