如何修复使用pg8000模块进行read_Sql时的编程错误

Question

嗨，我正试着用胶水来查询aws中的一个表，红移成熊猫的数据。我使用pg8000进行连接(因为aws胶水不支持sqlalchemy )。

当我使用熊猫的read_sql或read_sql_query函数查询表时，我在dataframe中得到了额外的char，我猜这是pg8000 dbapi的问题。

conn = pg8000.connect(user = 'postgres', password = '*****', host =127.0.0.1,  port = 5439, database = 'lifungdb')

cursor = conn.cursor()
df=pd.read_sql("select * from Customer",conn)
print(df)

print (df)返回列中的exta char b。如何去除多余的焦炭

b'id'  b'Name' b'Address' b'Contact
 1       Sam     Texas       na

Answer 1

使用list-comprehension解码utf-8字符串：

import pandas as pd
a = [['1', 'sam', 'Texas', 'na']]
df = pd.DataFrame(a, columns=[b'id', b'Name', b'Address', b'Contact'])
df.columns = [x.decode('utf-8') for x in df.columns]
print(df)

输出

  id Name Address Contact
0  1  sam   Texas      na