import pandas as pd import math import pymssql host = '127.0.0.1' def get_connection(host,user,pwd,db): conn = pymssql.connect(host,user,pwd,db)#返回Connections模块下的Connection实例 if conn: print(db,'连接成功') return conn def execuete_sql(sqlstr,conn)->pd.DataFrame: try: cursor = conn.cursor()#返回一个新的游标对象 cursor.execute(sqlstr) df = pd.DataFrame(cursor.fetchall()) return df except Exception as e: print(e) conn.close() a=get_connection('127.0.0.1','sa','1a2b3c4D','HYCX.Noise.ShunDe') b=execuete_sql('SELECT Id , Name FROM [HYCX.Noise.ShunDe].[dbo].[T_Road]',a) print(b) 我要怎么把b返回的df中的columns改掉
时间: 2024-04-13 16:29:17 浏览: 117
要修改DataFrame `b`中的列名,你可以使用`df.rename()`方法来实现。`rename()`方法接受一个字典作为参数,该字典包含需要修改的列名和对应的新列名。
以下是修改后的代码示例:
```python
import pandas as pd
import math
import pymssql
host = '127.0.0.1'
def get_connection(host,user,pwd,db):
conn = pymssql.connect(host,user,pwd,db)#返回Connections模块下的Connection实例
if conn:
print(db,'连接成功')
return conn
def execuete_sql(sqlstr, conn) -> pd.DataFrame:
try:
cursor = conn.cursor()#返回一个新的游标对象
cursor.execute(sqlstr)
df = pd.DataFrame(cursor.fetchall())
return df
except Exception as e:
print(e)
conn.close()
a = get_connection('127.0.0.1','sa','1a2b3c4D','HYCX.Noise.ShunDe')
b = execuete_sql('SELECT Id , Name FROM [HYCX.Noise.ShunDe].[dbo].[T_Road]', a)
# 修改列名
b = b.rename(columns={0: 'ID', 1: '名称'})
print(b)
```
在上述代码中,我们使用`rename()`方法将列名 `0` 修改为 `ID`,将列名 `1` 修改为 `名称`。然后我们打印出修改后的DataFrame `b` 的内容。你可以根据需要修改字典中的键值对来更改其他列的名称。
阅读全文