Python实现CSV格式文件批量导入至MySQL

一、建立连接

config = dict(host='localhost', user='root', password='root', db='db1', charset='utf8', cursorclass=pymysql.cursors.DictCursor)
self.conn = pymysql.connect(**config)
self.cursor = self.conn.cursor()

二、获取CSV格式文件位置

self.dir = 'T:/stock_data/'
files_list = os.listdir(self.dir)
return files_list

三、用pandas读取文件

for file in files_list:
    # pandas读取文件,skipinitialspace用来跳过' '
    df = pd.read_csv(self.dir + file, encoding='gbk', skipinitialspace=True)
    filename = os.path.split(file)[-1].split('.')[0].replace(' ', '_')
    self.csv2mysql(filename=filename, df=df)

四、将CSV文件里数据的格式转变为mysql可以使用的格式

columns = df.columns.tolist()  # 获取表格数据内的列标题文字数据
types = df.dtypes  # 获取文件内数据格式
field = []  
table = []
for item in range(len(columns)): 
    # 必须加上'`'这个点，否则在写入mysql是会报错
    item1 = '`' + columns[item] + '`'  
    if 'object' == str