快捷導(dǎo)航

Python+Matplotlib繪制小提琴圖的示例代碼

更新時(shí)間：2023年06月07日 08:30:23 作者：SpikeKing

小提琴圖 (Violin Plot) 類(lèi)似紡錘，是一種用來(lái)顯示數(shù)據(jù)分布和概率密度的圖形，本文為大家介紹了Matplotlib繪制小提琴圖的函數(shù)源碼，需要的可以參考一下

小提琴圖 (Violin Plot) 類(lèi)似紡錘，小提琴圖是一種用來(lái)顯示數(shù)據(jù)分布和概率密度的圖形，結(jié)合了箱線(xiàn)圖和核密度圖的特點(diǎn)。小提琴圖的中間部分是一個(gè)箱線(xiàn)圖，顯示了數(shù)據(jù)的中位數(shù)、四分位數(shù)和異常值。小提琴圖的兩側(cè)是一個(gè)核密度圖，顯示了數(shù)據(jù)的分布形狀。小提琴圖可以用來(lái)比較不同類(lèi)別或分組的數(shù)據(jù)，展示數(shù)據(jù)的差異和相似性。

示例如下：

源碼如下：

#!/usr/bin/env python
# -- coding: utf-8 --
"""
Copyright (c) 2022. All rights reserved.
Created by C. L. Wang on 2023/6/6
"""
import os

import matplotlib.pyplot as plt
import pandas as pd
import seaborn as sns

from myutils.project_utils import read_excel_to_df
from root_dir import DATA_DIR


def draw_violin_plot(
  df, score_col_name, label_col_name,
  x_label="", y_label="", title="",
  is_show=False, save_name=""):
  """
  繪制小提琴圖
  :param df: 數(shù)據(jù)格式，至少包括兩列，即數(shù)據(jù)值列score，標(biāo)簽列l(wèi)abel，相同類(lèi)別的標(biāo)簽相同
  :param score_col_name: 數(shù)值列名稱(chēng)
  :param label_col_name: 標(biāo)簽列名稱(chēng)
  :param x_label: 顯示的x標(biāo)簽
  :param y_label: 顯示的y標(biāo)簽
  :param title: 顯示的圖名稱(chēng)
  :param is_show:  是否IDE顯示
  :param save_name: 是否存儲(chǔ)文件，tight格式
  :return:
  """
  plt.figure(figsize=(16, 10), dpi=80)
  sns.violinplot(x=label_col_name, y=score_col_name, data=df, inner='quartile')

  plt.xlabel(xlabel=x_label)
  plt.ylabel(ylabel=y_label)

  if title:
    plt.title(title, fontsize=12)
  if save_name:
    # transparent=True
    plt.savefig(save_name, bbox_inches='tight', format='png')
  if is_show:
    plt.show()

def main():
  df = read_excel_to_df(os.path.join(DATA_DIR, "ourbest_20230601_tmscore_56.xls"))
  # df = read_excel_to_df(os.path.join(DATA_DIR, "ourbest_20230605_dockq_9_final.xls"))
  # df.info()

  # 加工數(shù)據(jù)格式，滿(mǎn)足數(shù)值和標(biāo)簽的格式
  score1 = df["m0-score"]
  score2 = df["m1-score"]
  score3 = df["cur-score"]
  label_col = ["SOTA" for _ in range(df["m0-score"].size)] + \
              ["Baseline" for _ in range(df["m1-score"].size)] + \
              ["Our Best" for _ in range(df["cur-score"].size)]
  score_list = pd.concat([score1, score2, score3])
  score_name = "DockQ"
  data = {score_name: score_list, "label": label_col}
  new_df = pd.DataFrame(data)
  # df.info()
  draw_violin_plot(new_df, score_col_name=score_name, label_col_name="label",
                   x_label="", y_label=score_name,
                   is_show=True, save_name="xxx.png")


if __name__ == '__main__':
  main()

到此這篇關(guān)于Python+Matplotlib繪制小提琴圖的示例代碼的文章就介紹到這了,更多相關(guān)Python Matplotlib繪制小提琴圖內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: