熊貓讀取SQL到CSV - Int轉換

Question

我有數據需要上傳到S3然後複製到Redshift的本地Postgres數據庫。

要做到這一點，我使用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') 

，它執行的varchar(255), varchar(255), int, int返回行的SQL。但是，由於其中一些值可能爲空，所以我遇到了Pandas Caveat about integers and NaN values。

This post通過將na設置爲任意int並將astype顯式設置爲int來解決它們的問題。他們能夠這樣做是因爲它們的所有列都具有相同的數據類型。但是，我有一個varchar（255）和int的混合。

我該如何強制df.to_csv(...)輸出我的整數作爲整數（而不是浮點數）？

感謝，

Answer 1

你爲什麼不能插入這些上面兩行df.to_csv()？

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