绘制瀑布图思路:遍历指定文件目录下所有的csv文件,每读一个文件,取文件前20行数据进行保存,如果超过规定的行数300行,将最旧的数据删除,仅保留300行数据进行展示。
网上找的大部分绘制瀑布图的代码,均无法呈现动态效果。不是动态的效果还能称为瀑布图吗?(疑问),博主自己摸索了一套绘制动态图的思路,这里全部代码贴出来,如果需要绘制用的数据,可以私信。当然也可以改代码以适用自己的数据。
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()