python按钮读写csv文件

版权声明：添加我的微信wlagooble，开启一段不一样的旅程 https://blog.csdn.net/nineship/article/details/84138671

from Tkinter import *
import cv2
import matplotlib.pyplot as plt
import matplotlib.image as mpimg
import os
import shutil
import csv

file_name_list = []
csv_list = []

now_id = 0
dic_name = ""

csv_name_list = []
csv_value_list = []
csv_save_list = []

def is_normal():
    global now_id
    if now_id < 0:
        now_id = 0
    if now_id >= len(file_name_list):
        now_id = len(file_name_list)-1
def init():
    global dic_name
    dic_name = raw_input("dict name: ")
    plt.ion()

    for root, dirs, files in os.walk("./clip/"+dic_name+"/", topdown=False):
        for name in files:
            print (name)
            file_name_list.append(name)
    read_csv("./csv/box_key.csv")
def up_pic():
    global now_id
    is_normal()
    name = "clip/"+dic_name+"/"+file_name_list[now_id]
    print (name)
    I = mpimg.imread(name)
    plt.imshow(I)
    plt.show()
    now_id -= 1
def next_pic():
    global now_id
    is_normal()
    name = "clip/"+dic_name+"/"+file_name_list[now_id]
    print (name)
    I = mpimg.imread(name)
    plt.imshow(I)
    plt.show()
    now_id += 1

def move_error():
    shutil.move("./clip/"+dic_name+"/"+file_name_list[now_id],"./error/"+dic_name)
    file_name_list.pop(now_id)
    next_pic()

def del_csv(csv_reader,line):
    csv_reader.drop(line)
    #csv_reader.close()

    return True
def find_csv(jpg_name):

    return index

csv_reader = csv.reader(open('./csv/box_key.csv','r'))
for row in csv_reader:
        csv_list.append(row)
        csv_name_list.append(row[0])
        csv_value_list.append(row[1:])
#print ("csv_name_list:",csv_name_list)
#input()
def find_error():
    init()

    root = Tk()

    Button(root, text='up', width=20,height=10,command=up_pic).pack()
    Button(root, text='next', width=20,height=10,command=next_pic).pack()
    Button(root, text='move', width=20,height=10,command=move_error).pack()

    root.mainloop()
def find_list(value):
    index = -1
    for i in range(len(csv_name_list)):
        if csv_name_list[i].find(value) != -1:
            index = i
            break
    return index

#print (csv_name_list)
#print (csv_value_list)

for root, dirs, files in os.walk("./error/", topdown=False):
        for name in files:
            #print (name)
            index = find_list(name)
            if index != -1:
                continue

            print ("index:",index)
            #print ("csv_list:",csv_list[index])
            #del_csv(csv_reader,index)            
            #csv_list.pop(index)
            csv_save_list.append(csv_list[index])
            #input()

csvfile = open("./csv/csv_save/csv_save.csv", 'w')
csvwrite = csv.writer(csvfile)
csvwrite.writerow(csv_save_list)
csvfile.close()
print ("success")