Pandas Чтение SQL в CSV-Int-преобразование

Question

У меня есть локальная база данных Postgres, которую мне нужно загрузить на S3, а затем скопировать в Redshift.

Для этого я использую Python панда следующим образом:

engine = create_engine(self.engine) 
connection = engine.raw_connection() 
df = pd.read_sql(<sql string>, connection, coerce_float=False) 
df.to_csv(<output fn>, header=True, index=False, encoding='utf-8') 

SQL, что он выполняет возвращает строку varchar(255), varchar(255), int, int. Однако, поскольку некоторые из этих значений могут быть нулевыми, я запускаю Pandas Caveat about integers and NaN values.

This post решает проблему, устанавливая na произвольным int и астерически явно как int. Они могут это сделать, потому что все их столбцы имеют один и тот же тип данных. Однако у меня есть сочетание varchar (255) и int.

Как я могу заставить df.to_csv(...) вывести мои ints как int (а не плавает)?

Спасибо,

Answer 1

Почему вы не можете вставить эти две строки выше df.to_csv()?

df = df.fillna(<some_int>) 
df[[int_col1, int_col2]] = df[[int_col1, int_col2]].astype('int64')