3

我正在编写一个脚本来在 csv 文件中写入查询结果行。数据是电子邮件地址，如下所示：

email1@mail.com

email2@mail.com

email3@mail.com

现在，我的代码将结果写入 csv 文件，如下所示：

('email1@mail.com'), ('email2@mail.com'), ('email3@mail.com'),

如何使用下一行而不是行？

编辑：似乎我的查询给了我括号之间的数据，例如（'email1@mail.com'），

这是代码：

import pymysql
import csv

db = pymysql.connect(host="localhost", port=3306, user="foo", passwd="bar", db="db")
cursor = db.cursor()

query3 = """SELECT email FROM tmp"""


cursor.execute(query)

data = cursor.fetchall()
list = []

for row in data :
  value = str(row)
  list.append(value)    
  file = open('file.csv', 'wb')
  data = csv.writer(file)
  data.writerow(list)

cursor.close()
db.close()
file.close()
4

2 回答 2

2

以下看起来可以解决您的问题：

f = csv.writer(open("file.csv", "w"))
for row in data:
    f.writerow([str(row)])

对于更详细/适当的解决方案，您必须向我们提供更多详细信息，例如您的输入格式是什么样的，以及您希望输出文件是什么样的（实际数据，而不是抽象的“a，b，c ”）。

作为旁注，不要调用您的变量list或file. 这些名称会影响内置列表/文件类型。

编辑：如果您所做的只是每行写一个电子邮件地址，那么您实际上并不是在使用 CSV，也不需要该模块。这是我要做的：

f = open("file.txt", "w")
for i in email_addresses:
    f.write(i + "\n")
于 2014-03-04T21:35:14.083 回答
1

像 pymysql 这样的模块符合 Python DB API。游标的fetchall()方法返回一个元组列表。每个元组代表数据库查询中的一行。将其转换为字符串可能无法满足您的需求：

>>> str((1, ))
'(1,)'

该writerow方法期望给定一个序列，并且该序列的每个元素都作为行中的一个值输出，并根据所选格式具有适当的表示和分隔符。

您当前的逻辑循环遍历数据库查询返回的所有行。每次循环时，它都会将另一个（可能不正确的）值添加到列表中，打开一个文件进行写入，并写出当前列表。由于每次迭代都会创建一个新文件，因此您只会看到上次迭代中写入的内容。

这是可能做你想做的代码。

import csv

data = [("first",), ("second",), ("Third",)]
list = []

file = open('file.csv', 'wb')
writer = csv.writer(file)
for row in data :
    writer.writerow(row)
file.close()
于 2014-03-04T22:24:56.823 回答