我是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个外观相似的表,并且仅进行少量更改就重复相同的代码是没有意义的。在这种情况下可以使用循环吗?
谢谢。