import pandas as pd import matplotlib.pyplot as plt # 划分区间 bins = [0, 1, 2, 3, 4, 5, 10, 15, 20, 25, 30, 35, 40, 45, 50, 60, 70, 80, 90, 100, 1220] # 获取数据 data1 = data_forecast_is_stk_bs_rpt_zq_befor_big['本年比上一年归母倍数'] data2 = data_forecast_is_stk_bs_rpt_cw_befor_big['本年比上一年归母倍数'] # 统计每个柱子的数据 hist1, edges1 = pd.cut(data1, bins=bins, right=False, include_lowest=True, retbins=True) hist2, edges2 = pd.cut(data2, bins=bins, right=False, include_lowest=True, retbins=True) counts1 = hist1.value_counts() counts2 = hist2.value_counts() # 按照 bins 的顺序给每个柱子指定标签 labels = counts1.index.astype(str) # 创建 Figure 对象和两个子图对象 fig, (ax1, ax2) = plt.subplots(nrows=1, ncols=2, figsize=(18, 6)) # 在第一个子图上绘制柱形图 ax1.bar(labels, counts1) for i, v in enumerate(counts1.values): ax1.text(i, v, str(v), ha='center', va='bottom') ax1.set_title('准确数据-本年小于前一年归母倍数(证券)') ax1.set_xlabel('Range') ax1.set_ylabel('Count') # 在第二个子图上绘制柱形图 ax2.bar(labels, counts2) for i, v in enumerate(counts2.values): ax2.text(i, v, str(v), ha='center', va='bottom') ax2.set_title('准确数据-本年小于前一年归母倍数(财务)') ax2.set_xlabel('Range') ax2.set_ylabel('Count') # 调整子图的位置和间距 fig.tight_layout() # 显示图表 plt.show() 是左右2个图要合成一个对比图
时间: 2023-12-09 14:05:35 浏览: 109
可以将两个子图合并成一个对比图,方法如下:
1. 首先将创建 Figure 对象的代码修改为 `fig, ax = plt.subplots(nrows=1, ncols=2, figsize=(18, 6))`,这样就只创建一个 Figure 对象和一个包含两个子图的 ax 对象。
2. 然后在第二个子图上绘制柱形图时,将 `ax2.bar(labels, counts2)` 改为 `ax.bar(labels, counts2, alpha=0.5)`,这样就可以在同一个子图上绘制两个柱形图,并使它们半透明重叠显示。
3. 最后在第一个子图上添加一个图例,代码如下:
```
ax1.bar(labels, counts1, label='证券', alpha=0.5)
ax1.legend()
```
这样就可以给第一个柱形图添加一个图例,用以表示该柱形图代表的是证券数据。
完整代码如下:
```
import pandas as pd
import matplotlib.pyplot as plt
# 划分区间
bins = [0, 1, 2, 3, 4, 5, 10, 15, 20, 25, 30, 35, 40, 45, 50, 60, 70, 80, 90, 100, 1220]
# 获取数据
data1 = data_forecast_is_stk_bs_rpt_zq_befor_big['本年比上一年归母倍数']
data2 = data_forecast_is_stk_bs_rpt_cw_befor_big['本年比上一年归母倍数']
# 统计每个柱子的数据
hist1, edges1 = pd.cut(data1, bins=bins, right=False, include_lowest=True, retbins=True)
hist2, edges2 = pd.cut(data2, bins=bins, right=False, include_lowest=True, retbins=True)
counts1 = hist1.value_counts()
counts2 = hist2.value_counts()
# 按照 bins 的顺序给每个柱子指定标签
labels = counts1.index.astype(str)
# 创建 Figure 对象和一个包含两个子图的 ax 对象
fig, ax = plt.subplots(nrows=1, ncols=2, figsize=(18, 6))
# 在第一个子图上绘制柱形图,并添加图例
ax1 = ax[0]
ax1.bar(labels, counts1, label='证券', alpha=0.5)
ax1.legend()
for i, v in enumerate(counts1.values):
ax1.text(i, v, str(v), ha='center', va='bottom')
ax1.set_title('准确数据-本年小于前一年归母倍数(证券)')
ax1.set_xlabel('Range')
ax1.set_ylabel('Count')
# 在第二个子图上绘制柱形图
ax2 = ax[1]
ax.bar(labels, counts2, label='财务', alpha=0.5)
for i, v in enumerate(counts2.values):
ax2.text(i, v, str(v), ha='center', va='bottom')
ax2.set_title('准确数据-本年小于前一年归母倍数(财务)')
ax2.set_xlabel('Range')
ax2.set_ylabel('Count')
# 调整子图的位置和间距
fig.tight_layout()
# 显示图表
plt.show()
```
这样就可以得到一个包含两个柱形图的对比图,其中一个柱形图代表证券数据,另一个柱形图代表财务数据。
阅读全文