python - Pandas to_sql 设置列类型

Question

我有一个字符串格式的数字列，我想将它作为 float 发送到 PostresSQL。如何确保 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()}
)