优化代码以从不同的数据框架创建条形图

时间:2018-11-17 09:40:15

标签: dataframe optimization graph

我是Python的新手,我正在学习根据数据框创建图表。我有几个看起来像这样的数据框:

Company Case 1 Case 2 Case 3 Max value Min value
  A      -15     25     28      28       -15
  B       20     26     21      26        21
  C       18    -22     10      22       -22

Company Case 1 Case 2 Case 3 Max value Min value
  A      -15     28     26      28       -15
  B       12     26     11      26        11
  C       12    -25     10      12       -25

前三列仅是一些值,后两列分别对应于每个公司的最大和混合值。我为每个数据框的条形图创建了两个代码。代码如下:

graph_A = df1[['Case 1', 'Case', 'Case 3']].plot(kind = 'bar', width=.33, legend = True, color=['green', 'blue', 'red'], alpha = .7) plt.suptitle("Scenario 1") graph_C.yaxis.set_major_formatter(mpl.ticker.StrMethodFormatter('{x:,.0f}')) plt.legend(frameon = False) plt.xticks(rotation = 0) plt.xlabel("Company name") plt.ylabel("Value")

graph_B = df2[['Case 1', 'Case', 'Case 3']].plot(kind = 'bar', width=.33, legend = True, color=['green', 'blue', 'red'], alpha = .7) plt.suptitle("Scenario 2") graph_C.yaxis.set_major_formatter(mpl.ticker.StrMethodFormatter('{x:,.0f}')) plt.legend(frameon = False plt.xticks(rotation = 0) plt.xlabel("Company name") plt.ylabel("Value")

我的问题是如何优化代码?例如,假设我有100个外观相似的表,并且仅进行少量更改就重复相同的代码是没有意义的。在这种情况下可以使用循环吗?

谢谢。

0 个答案:

没有答案