首页 > 编程笔记 > Python笔记 阅读:18

Visdom bar()绘制条形图(附带实例)

Visdom 是 PyTorch 深度学习最常用的模型可视化工具。Visdom 提供了多种绘图函数,可以用于实现数据的可视化。

bar() 函数用于绘制规则的、堆叠的或分组的条形图。它接受一个 N 或 N×M 张量 X 作为输入,它指定了每个条的高度。如果 X 包含 M 列,则对每一行对应的值进行堆叠或分组(取决于 opts.stacked 的选择方式)。除 X 外,还可以指定一个(可选的)N 张量 Y,它包含相应的 X 轴的值。

以下是该函数目前支持的选项:
以下是实现一个条形图的代码:
# 导入相关库
import visdom
import numpy as np

# 使用 Visdom 库绘制条形图
vis.bar(
    # 条形图的数据,这里使用了随机数
    X=np.abs(np.random.rand(4, 3)),
    # 图表选项
    opts=dict(
        stacked=True,                                # 条形图堆叠
        legend=['低价值客户', '一般价值客户', '高价值客户'],  # 图例
        rownames=['2017', '2018', '2019', '2020']    # 行名
    )
)
输出如下图所示:


图 1 条形图

相关文章