从csv中读数据，并写到csv中

zn505119020

于 2017-08-01 17:52:13 发布

阅读量955

点赞数

CC 4.0 BY-SA版权

分类专栏：工作相关

本文链接：https://blog.youkuaiyun.com/zn505119020/article/details/76549069

工作相关专栏收录该内容

136 篇文章

订阅专栏

本文介绍了一个使用 Python 从 MySQL 数据库中的 'feedback' 表读取数据，并将这些数据写入到 CSV 文件的过程。该脚本适用于批量处理 eBay 编号对应的反馈数据，通过连接指定的数据库并执行 SQL 查询来获取所需的信息。

import pymysql
import numpy as np
import csv
def connectViogidata(ebayno):
    db = pymysql.connect(host='', user='s', passwd='2', db='viogidata', port=3306, charset='utf8')
    cursor = db.cursor()
    sql = ' select  id, ebayno, product_name, price, currency, when_time,shop_name from feedback where ebayno = %s'
    cursor.execute(sql,(ebayno))
    db.commit()
    cursor.close()
    db.close()
    result = cursor.fetchall()
    return result

if __name__ == '__main__':
    # 写入到csv表中
    out = open("feedback_WM-D39S0106.csv","w",newline="")
    csv_writer = csv.writer(out)
    csv_writer.writerow(["id","ebayno","product_name","price","currency","when_time","shop_name"])

    #从csv文件中读取数据
    with open("1.csv",newline="") as f:
        reader = csv.reader(f)
        k = 0
        for row1 in reader:
            if k  == 0:  #去掉列名
                k = 1
                continue
            if len(row1) == 0:
                break
            ebayno = row1[0]
            feedback = connectViogidata(ebayno)  # 得到元组数据
            if feedback is not None:
                for row in feedback:  # 去掉外层括号
                    csv_writer.writerow([str(row[0]) , str(row[1]) , str(row[2]),str(row[3]) , str(row[4]), str(row[5]) , str(row[6]) ])