python读取数据并画图

import numpy as np
import matplotlib.pyplot as plt
import os
import time


def getFile(path):  # 获取目录下数据文件的数据
    datalist = []
    for root, dirs, files in list(os.walk(path)):  # root为str文件夹路径，dirs、files为lists，文件夹名字，文件名字
        for i in files:
            if i.endswith('.txt'):
                datalist.append(np.loadtxt(i))
    datalist = np.array(datalist)
    return datalist


def im_plot(datalis):
    x = np.linspace(0, 10, 1000)
    o = []
    o.append(x)
    o.append(x)
    o.append(x)
    o = np.array(o)
    plt.plot(o.T, datalis.T, color="black", linewidth=1)


if __name__ == "__main__":
    startTime = time.time()
    filePath = "D:\\16\\15"
    plt.figure(2, figsize=(8, 4))