记第一次用python写界面

花了两三个小时学了Tkinter，做了一个将数据绘制成图的小工具。

1. 获取路径下的所有文件or获取路径下指定名称的文件

1.1  打开文件

//1. 用来放文本框中的文字
filename = StringVar()
//2. 新建一个单行文本框
txtDataPath = tkinter.Entry(window,width=60,textvariable=filename).grid(row=0,column = 1)

def OpenFile():
    file = filedialog.askopenfilename()
    if(file):
      filename.set(file)//3. 打开文件后把文件名写在文本框中

1.2 打开文件夹

选数据的时候要打开文件夹，存图的时候也要，所以if...else...

def OpenDirectory(IsOpenOrSave):
    directory = filedialog.askdirectory()
    if(directory):
        if(IsOpenOrSave == 0):
            filename.set(directory)
        else:
            savepath.set(directory)

1.3 获取路径下指定名称的文件

def listSameFile(dir):
    str = "Whole"
    for x in os.listdir(dir):
        path = os.path.join(dir, x)
        if os.path.isfile(path) and str in os.path.splitext(x)[0]:
            split_data(path)
        if os.path.isdir(path):
            listSameFile(path)

2. 读取数据进行切割

数据生成的图片将用于深度学习，因此越多越好。两张图片的数据有固定的交集。

代码：

def split_data(data,stride, size):
    length = data.shape[0]
    start = 0
    end = start + size
    while(start<length-size):
        print(start)
        print(end)
        start+=stride
        end = start+size
        if(end>length):
            end = length
            start = length-size
            print(start)
            print(end)

写的很烂，绞尽脑汁也没有想到可以只打印一次

3. 绘图

绘图时我希望可以直接保存不要画出来，但是

代码必须是先plt.savefig(‘save.jpg’)再plt.show()才能够保存显示的图像

如果先plt.show()再plt.savefig('save.jpg')图像保存的是白色空白图

如果去掉plt.show()只保存图像plt.savefig('save.jpg')，单幅图像可以，如果循环文件夹里的图像，保存的图像会叠加，我白把数据切开了。

加了一个close()后：(忘记哪条数据了，换了条数据。。。

切片+画图：

def split_data(filepath):
    data = pd.read_csv(filepath, sep=' ')
    size = data.shape[0]
    #print(size)
    len = int(datalength.get())
    stride = int(datastride.get())
    start = 0
    end = start + len

    while(end<size):
        plot_data(data,start,end)
        start += stride
        end = start + len
    end = size
    start = end-len
    plot_data(data,start,end)

def plot_data(data,start,end):
    i = pic_index.get()
    i += 1
    pic_index.set(i)
    #print(i)
    #print(start)
    #print(end)
    data_cut = data[start:end]
    ax = plt.gca()
    ax.yaxis.set_major_locator(MultipleLocator(0.01))
    plt.plot(data_cut, color = 'black')
    if (IsShowCoordinate.get() == 0):
        plt.axis('off')
    if (IsSavePic.get() == 1):
        plt.savefig(savepath.get() + "/{}.png".format(i))
        plt.close()
    else:
        plt.show()
　　　　 plt.close()

4. 全代码

import tkinter
from tkinter import *
import os
from tkinter import filedialog
import pandas as pd
from matplotlib import pyplot as plt
from matplotlib.ticker import MultipleLocator

LOGO_PATH = "./fish.ico"
window = tkinter.Tk()
window.title("数据转化为图片")
window.iconbitmap(LOGO_PATH)
window.geometry("700x200")
window.maxsize(1000, 1200)
# ----------------------------------
filename = StringVar()
datastride = StringVar()
datalength = StringVar()
savepath = StringVar()

lblDataPath = tkinter.Label(window, text="数据路径").grid(row=0,column = 0)
lblSavePath = tkinter.Label(window, text="存图路径").grid(row=1,column = 0)
lblDataLength = tkinter.Label(window, text="数据长度").grid(row=2,column = 0)
lblDataStride = tkinter.Label(window, text="滑动长度").grid(row=3,column = 0)

txtDataPath = tkinter.Entry(window,width=60,textvariable=filename).grid(row=0,column = 1)
txtSavePath = tkinter.Entry(window,width=60,textvariable = savepath).grid(row=1,column = 1)
txtDataLength = tkinter.Entry(window,width=10,textvariable = datalength).grid(row=2,column = 1)
txtDataStride = tkinter.Entry(window,width=10,textvariable = datastride).grid(row=3,column = 1)

IsShowCoordinate = IntVar()
IsSavePic = IntVar()
ckbCoordinate = Checkbutton(window, text = "显示坐标", variable = IsShowCoordinate).grid(row=5,column = 3)
ckbSavePic= Checkbutton(window, text = "保存图片", variable = IsSavePic).grid(row=6,column = 3)

pic_index = IntVar()
IsDrawOnce = IntVar()

def OpenFile():
    IsDrawOnce.set(0)
    file = filedialog.askopenfilename()
    if(file):
      filename.set(file)

def OpenDirectory(IsOpenOrSave):
    directory = filedialog.askdirectory()
    if(directory):
        if(IsOpenOrSave == 0):
            filename.set(directory)
        else:
            savepath.set(directory)

def split_data(filepath):
    i = pic_index.get()
    data = pd.read_csv(filepath, sep=' ')
    len = int(datalength.get())
    stride = int(datastride.get())
    start = 0
    end = start + len
    size = data.shape[0]

    arr = []
    while(end<size):
        i+=1
        pic_index.set(i)
        data_cut = data[start:end]
        ax = plt.gca()
        ax.yaxis.set_major_locator(MultipleLocator(0.01))
        plt.plot(data_cut,color='black')
        if(IsShowCoordinate.get() == 0):
            plt.axis('off')
        if(IsSavePic.get() ==1):
            plt.savefig(savepath.get()+"/{}.png".format(i))
            plt.close()
        else:
            plt.plot(data_cut)
            plt.show()
        start+=stride
        end = start+len

def listSameFile(dir):
    str = "Whole"
    for x in os.listdir(dir):
        path = os.path.join(dir, x)
        if os.path.isfile(path) and str in os.path.splitext(x)[0]:
            split_data(path)
        if os.path.isdir(path):
            listSameFile(path)

def drawing():
    listSameFile(filename.get())

def draw_one_pic():
    split_data(filename.get())

def btn_OpenFile_click():
    IsDrawOnce.set(0)
    OpenFile()

def btn_OpenDirectory_click():
    IsDrawOnce.set(1)
    OpenDirectory(0)

def btn_SaveDirectory_click():

    OpenDirectory(1)

def btn_draw_click():
    pic_index.set(0)
    if(IsDrawOnce.get()==0):
        draw_one_pic()
    else:
        drawing()

Button(window, text="打开文件", command=btn_OpenFile_click).grid(row=0,column = 4)
Button(window, text="打开文件夹", command=btn_OpenDirectory_click).grid(row=0,column = 5)
Button(window, text="选择路径", command=btn_SaveDirectory_click).grid(row=1,column = 4)
Button(window, text="绘制", command=btn_draw_click).grid(row=5,column = 4)

window.mainloop()

写的实在烂，堪堪运行。

想要设置一个变量以命名图片的名字，不会。。。

控件不会靠左。。。（已解决）

新代码更新在 https://gitee.com/yurj0403/line1.git

生成exe后，ico总是会报错，只好把ico去掉了。