【Python】Python快速绘制dot_heatmap矢量图

KmBase

已于 2022-12-22 19:36:11 修改

阅读量625

点赞数

CC 4.0 BY-SA版权

分类专栏： # Python 文章标签： python matplotlib 开发语言

于 2022-12-05 10:47:54 首次发布

本文链接：https://blog.csdn.net/qq_25262697/article/details/128182711

Python 专栏收录该内容

18 篇文章

订阅专栏

本文介绍如何使用milkviz库中的dot_heatmap函数来生成点热图，并通过matplotlib将图表保存为SVG格式。文章提供了完整的示例代码及运行结果。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

一、说明

基于milkviz库，使用matplotlib的savefig接口保存为svg矢量格式，以下为生成效果。
在这里插入图片描述

二、安装

milkviz库是针对matplotlib的二次封装，以下官网基本示例，可快速生成以下数据类型图：
在这里插入图片描述
使用pip安装即可：
pip install milkviz

三、源代码

#!/usr/bin/env python
# -*- encoding: utf-8 -*-
'''
@File    :   plot_dot_heatmap.py
@Time    :   2022/12/05 10:39:32
@Author  :   KmBase
@Version :   1.0
@License :   (C)Copyright 1984-2022, KmBase
@Desc    :   None
'''


"""
Dot heatmap Example
======================

Here shows how to draw dot heatmap

"""
import numpy as np
import milkviz as mv
from matplotlib import pyplot as plt

np.random.seed(0)
shape = (15, 15)
colors = np.random.randint(1, 100, shape)
sizes = np.random.randint(1, 100, shape)
matrix = np.random.randint(1, 100, shape)
labels = [
    "Apple", "Avocado", "Banana", "Blueberries", "Coconut", "Kiwifruit",
    "Lemon", "Mango", "Olives", "Papaya", "Persimmon", "Plum", "Quince",
    "Soursop", "Watermelon"
]
mv.dot_heatmap(sizes,
               colors,
               matrix,
               xticklabels=labels,
               dot_size_legend_kw={"title": "Dot Size"},
               dot_hue_cbar_kw={"title": "Dot Color"},
               matrix_cbar_kw={"title": "Matrix Color"})
plt.tight_layout()
plt.savefig('./0.svg')
plt.show()