Viết khung dữ liệu gấu trúc nhanh chóng vào postgres

Cách tiếp cận thứ hai của bạn sẽ rất nhanh.

Có hai vấn đề với mã của bạn:

1. Sau khi ghi csv vào f bạn được đặt ở cuối tệp. Bạn cần đặt lại vị trí của mình từ đầu trước khi bắt đầu đọc.
2. Khi viết csv, bạn cần bỏ qua tiêu đề và chỉ mục

Đây là mã cuối cùng của bạn sẽ trông như thế nào:

import io
f = io.StringIO()
pd.DataFrame({'a':[1,2], 'b':[3,4]}).to_csv(f, index=False, header=False)  # removed header
f.seek(0)  # move position to beginning of file before reading
cursor = conn.cursor()
cursor.execute('create table bbbb (a int, b int);COMMIT; ')
cursor.copy_from(f, 'bbbb', columns=('a', 'b'), sep=',')
cursor.execute("select * from bbbb;")
a = cursor.fetchall()
print(a)
cursor.close()