写一个多柱状图带表格的pytorch代码
时间: 2023-06-30 07:04:08 浏览: 133
以下是一个简单的多柱状图带表格的 PyTorch 代码示例,使用了 Matplotlib 库来可视化数据:
```python
import torch
import matplotlib.pyplot as plt
# 生成随机数据
data1 = torch.randn(5)
data2 = torch.randn(5)
data3 = torch.randn(5)
# 设置图表参数
labels = ['Label1', 'Label2', 'Label3', 'Label4', 'Label5']
x = torch.arange(len(labels))
width = 0.25
# 绘制柱状图
fig, ax = plt.subplots()
rects1 = ax.bar(x - width, data1, width, label='Data1')
rects2 = ax.bar(x, data2, width, label='Data2')
rects3 = ax.bar(x + width, data3, width, label='Data3')
# 添加表格
table_data = [data1.tolist(), data2.tolist(), data3.tolist()]
table = ax.table(cellText=table_data, colLabels=labels, loc='bottom')
# 设置图表标题,标签和范围
ax.set_title('Multi-Bar Chart with Table')
ax.set_xticks(x)
ax.set_xticklabels(labels)
ax.legend()
ax.autoscale_view()
# 展示图表
plt.show()
```
这个示例代码将会生成一个包含三组数据的多柱状图和一个包含所有数据的表格。你可以根据需要修改数据和参数来适应你的场景。
阅读全文