csv 保存多个sheet

import pandas as pd
import MySQLdb   
import csv
import random
import pymysql
import pandas as pd
import MySQLdb as mdb
import sys
import time
import os
import numpy as np
import time
import gzip
import datetime


def fun(p_list):
      writer = pd.ExcelWriter('D:\\test.xlsx')
      for i in range(len(p_list)):
            if i>100:
                  return
            m_dir  = os.path.dirname(p_list[i]) 
            m_file = os.path.basename(p_list[i])

            print("m_dir = ",m_dir)
            print("m_file = ",m_file)

            file_name = os.path.basename(m_file)
            file_name = file_name.split('.')[0]
            print(file_name)

            data1 = pd.read_csv(p_list[i], encoding="gbk")
           
            data1.to_excel(writer,sheet_name=file_name, index=0)
           
            writer.save()
      writer.close()

def walkFile_list(file):
    p_list = []
    m_count=0
    for root, dirs, files in os.walk(file):
        for f in files:
            m = os.path.join(root, f)
            ss=os.path.splitext(m)
            a = os.path.basename(m)
            if (".csv" == ss[1]):
                p_list.append(m)
            else:
                pass
    return p_list


def main():
      dir = r"E:\\"
      m_list = walkFile_list(dir)
      fun(m_list)

if __name__ =="__main__":
      print("start")
      main()

你可能感兴趣的:(【Python】,python)