熊猫to_sql设置列类型

Question

我有一列要以PostresSQL浮点数形式发送给String格式的数字。如何确保SQLAlchemy将此列设置为float？（请注意，该列中可能是NaN）。这是代码

import pandas as pd
from sqlalchemy import create_engine

engine = create_engine('postgresql://{}:{}@{}:5432/{}'.format(USER, DB_PW, HOST, DB))

df = pd.DataFrame({'String2Number': ['0,2', '', '0,0000001']})

df.to_sql(name='TABLE_NAME', con=engine, index=False)

Answer 1

首先，您的数字应采用小数点格式，因此我们需要用逗号替换您的小数。接下来，您应该确保该to_sql函数将使用float，并且可以使用dtype参数来实现此目的，该参数可以在插入数据库时设置列类型（基于sqlalchemy types）。这里的代码：

import pandas as pd
from sqlalchemy import create_engine
from sqlalchemy.types import Float # note this import to use sqlalchemy Float type

engine = create_engine('postgresql://{}:{}@{}:5432/{}'.format(USER, DB_PW, HOST, DB))

df = pd.DataFrame({'String2Number': ['0,2', '', '0,0000001']})

# Replacing ',' to '.'
df['String2Number'] = df['String2Number'].apply(lambda x: str(x).replace(',', '.'))

# Set column type as SQLAlchemy Float
df.to_sql(
  name='TABLE_NAME',
  con=engine,
  index=False,
  dtype={'String2Number': Float()}
)