matplotlib绘制动态瀑布图-Toy模板网

这篇具有很好参考价值的文章主要介绍了matplotlib绘制动态瀑布图。希望对大家有所帮助。如果存在错误或未考虑完全的地方，请大家不吝赐教，您也可以点击"举报违法"按钮提交疑问。

matplotlib绘制动态瀑布图,python,matplotlib

绘制瀑布图思路：遍历指定文件目录下所有的csv文件，每读一个文件，取文件前20行数据进行保存，如果超过规定的行数300行，将最旧的数据删除，仅保留300行数据进行展示。

网上找的大部分绘制瀑布图的代码，均无法呈现动态效果。不是动态的效果还能称为瀑布图吗？(疑问)，博主自己摸索了一套绘制动态图的思路，这里全部代码贴出来，如果需要绘制用的数据，可以私信。当然也可以改代码以适用自己的数据。文章来源地址https://www.toymoban.com/news/detail-801095.html

import os
import numpy as np 
import pandas as pd 
import matplotlib.pyplot as plt
import matplotlib.colors as mcolors
from matplotlib.animation import FuncAnimation

fig, ax = plt.subplots()
yLimLen = 300
pastData = np.ones((0, 0))

def update(csvFilePathName):
    df = pd.read_csv(csvFilePathName, header = None)
    arr = df.values[:10]  

    global pastData
    if not pastData.any():
        pastData = arr
    else:
        arr = np.row_stack((pastData, arr))
        pastData = arr

    lenNum = arr.shape[0]
    print(f'lenNum:{lenNum}')
    if lenNum > yLimLen:
        arr = arr[lenNum - yLimLen:-1]

    colors = ['greenyellow', 'yellow', 'darkblue', 'royalblue', 'orange', 'red']
    clrmap = mcolors.LinearSegmentedColormap.from_list('mycmap', colors)
    csvFileName =  csvFilePathName.split('/')[2]
    ax.set_xlabel(csvFileName)
    plt.imshow(np.flipud(arr), cmap = clrmap) #用于将数据反转，最上面是最新的数据


def configSet(drawUseFileNum = 50):
    path = r'./2023-09-28_15/'
    files = os.listdir(path)[:drawUseFileNum]

    filePathName = [path + i for i in files]
    print(f'一共有{len(files)}个文件')

    # fig, ax = plt.subplots()
    ax.set_ylim(0, yLimLen)
    # ax.set_axis_off()
    ax.invert_yaxis()
    ani = FuncAnimation(fig, update, frames = filePathName, 
            interval = 50, blit = False, repeat = False)
    # plt.show()
    ani.save('move.gif', writer='imagemagick')
    print(f'finish!')

if __name__ == '__main__':
    configSet()