Python打印SQL查询指定列结果时如何将内容写入文件

你调用newfile.write()未达预期，通常是以下几个问题导致的：

• 未正确管理文件句柄：要么没有以写入模式打开文件，要么写入后没有执行刷新/关闭操作，内容没有实际写入磁盘
• Pandas显示配置作用域错误：你写的pd.option_context只包裹了read_sql读数据的逻辑，后续调用to_string()时已经退出了上下文，Pandas会用默认的截断规则，输出带省略号的不完整内容，和控制台打印的全量结果不一致
• 额外性能问题：当前SQL用SELECT *拉取全表所有列再在内存里筛选PartNumber，表数据量大时会浪费大量内存和查询时间，建议直接在SQL中指定需要查询的列。

方案1：和现有打印输出完全一致的写入 #

如果需要写入文件的内容和控制台打印的内容完全相同，把文件操作、字符串转换逻辑都放到显示配置的上下文内，用with语句自动管理文件连接，避免手动关文件的疏漏：

import sqlalchemy
import pandas as pd

# 建立数据库连接
testDBCon = sqlalchemy.create_engine('xxx')
# 优化SQL：仅查询需要的PartNumber列，降低查询开销
query = "SELECT PartNumber FROM testDB.dbo.PIESData;"

# 显示配置上下文覆盖所有用到to_string的逻辑
with pd.option_context('display.max_rows', None, 'display.max_columns', None):
    df = pd.read_sql(query, testDBCon)
    output_content = df.to_string()
    # 控制台打印
    print(output_content)
    # 写入文件，指定utf-8编码避免特殊字符报错，with块退出时自动关闭文件、落盘内容
    with open("part_number_result.txt", "w", encoding="utf-8") as out_file:
        out_file.write(output_content)

注意：Windows环境下打开文件必须显式指定encoding="utf-8"，否则默认用GBK编码，遇到特殊字符会直接抛出编码错误。

方案2：纯数据导出（不需要保留打印格式时用） #

如果不需要保留控制台打印的索引、列对齐等格式，只是需要把所有PartNumber值存到文件里，直接用Pandas原生的导出方法即可，不受显示配置限制，导出效率更高：

import sqlalchemy
import pandas as pd

testDBCon = sqlalchemy.create_engine('xxx')
query = "SELECT PartNumber FROM testDB.dbo.PIESData;"
df = pd.read_sql(query, testDBCon)

# 控制台全量打印
with pd.option_context('display.max_rows', None):
    print(df.to_string())

# 导出为纯文本文件，每行一个PartNumber，不带索引、不带表头
df["PartNumber"].to_csv("part_number_result.txt", index=False, header=False, encoding="utf-8")

这种方式导出的文件没有多余的格式字符，适合后续做批量数据处理、导入其他系统使用。

内容的提问来源于stack exchange，提问作者Edward Wynman