Question

我目前正在尝试打印雪花连接器查询的输出并将其写入csv，但是在执行此操作时遇到了问题。目前，我的代码示例如下：

query = '''SELECT TOP 10 * FROM TABLE;'''

conn = snowflake.connector.connect(user='XXXX',
                                   password='XXXX',
                                   account='XXXX')

query_output = conn.cursor().execute(query)

我是否可以打印查询的输出（例如，print(query_output)，然后像query_output.to_csv(path)一样将其写入csv？

Answer 1

You can do like below :
---------------------------------------------------------------

import csv
.....
.....
.....
rows=cur.fetchall()
column_names = [i[0] for i in cur.description]
fp = open('samplefile.csv', 'w')
myFile = csv.writer(fp, lineterminator = '\n')
myFile.writerow(column_names)
myFile.writerows(rows)
fp.close()
.....
.....

Answer 2

我强烈建议使用Pandas'DataFrames API。在其他几种数据转换和分析功能中，它具有出色的CSV读写器实现very flexible。

Snowflake的Python连接器还natively supports将结果转换为Pandas DataFrame对象。

下面提供了一个非常简单的改编示例（添加了1行）：

query = '''SELECT TOP 10 * FROM TABLE;'''

conn = snowflake.connector.connect(user='XXXX',
                                   password='XXXX',
                                   account='XXXX')

query_output = conn.cursor().execute(query)

## You may have to run: pip3 install --user pandas ##

query_output.fetch_pandas_all().to_csv("/path/to/write/table.csv")

默认CSV格式使用,作为分隔符，使用特定于操作系统的换行符（\n，Windows除外，使用\r\n），并使用最小引号（自动引号{{1 }}列中包含定界符）。

Answer 3

您可以使用它来打印结果。要将其保存在csv中，只需使用for循环将结果保存在数据帧中，然后使用df.to_csv

    cursor.execute(query)
    sql_result = cursor.fetchall()
    print(sql_result)     ## just print the results or use the for loop below to print the results row by row 
    for row in sql_result:
        print(row)

Python输出雪花连接器到CSV

3 个答案: