如何通过Python忽略SQL列名？移除列名处理代码后报错求助

首先咱们先捋清楚核心问题：那两行代码可不是可有可无的——它们在帮你修复/重命名查询返回的列名，你后续的代码大概率依赖处理后的列名来操作，删掉自然会触发找不到列的错误。

为啥这两行代码不能删？ #

看你的SQL查询里用了表别名b，pyodbc返回的DataFrame列名很可能是带b前缀的格式（比如b.tradeid），甚至部分驱动会返回字节串格式的列名（比如b'tradeid'）。那两行代码的作用就是：

1. 遍历列名列表，把每个元素里的b字符（或字节串的b标识）去掉
2. 将处理后的干净列名重新赋值给DataFrame

如果删掉它们，后续你用results['tradeid']这种方式访问列时，就会因为找不到对应列名（实际列名是b.tradeid或b'tradeid'）抛出KeyError之类的错误。

怎么实现“忽略SQL列名”？ #

如果你不想依赖SQL返回的原始列名，有几种实用办法：

1. 直接用索引访问数据

完全跳过列名，通过位置索引操作数据：

# 获取第一列的所有数据
first_col_data = results.iloc[:, 0]
# 获取第2行第3列的具体值
target_value = results.iloc[1, 2]

2. 手动指定自定义列名

不管SQL返回什么列名，直接给DataFrame设置自己想用的列名：

# 假设查询返回3列，自定义列名列表
custom_cols = ['交易ID', '产品名称', '交易金额']
results.columns = custom_cols
# 之后就可以用自定义列名访问
print(results['交易ID'])

3. 在SQL语句中直接重命名列

最稳妥的方式是从源头解决，在SQL里给列重命名，这样返回的列名就是你想要的，Python端无需额外处理：

SELECT 
    b.tradeid AS trade_id,
    b.prodname AS product_name,
    b.amount AS trade_amount
FROM X.Y a 
INNER JOIN X.X b ON a.tradeid = b.tradeid 
WHERE a.prod...

这样pyodbc返回的DataFrame列名就是trade_id、product_name，那两行处理列名的代码自然就可以删掉了。

额外提醒：你的SQL语句有语法错误！ #

注意你的SQL里写了两个WHERE，连接表的条件应该用ON，正确写法是：

SELECT b.* 
FROM X.Y a 
INNER JOIN X.X b ON a.tradeid = b.tradeid 
WHERE a.prod...

这个语法错误也可能是你遇到奇怪报错的原因之一，建议先修正这个问题！

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