使用to_sql将数据附加到pandas中已存在的表中

Question

我有以下数据框架

ipdb> csv_data
  country    sale        date  trans_factor
0   India  403171  12/01/2012             1
1  Bhutan  394096  12/01/2012             2
2   Nepal   super  12/01/2012             3
3  madhya  355883  12/01/2012             4
4   sudan     man  12/01/2012             5

截至目前我正在使用下面的代码在表中插入数据,如果表已经存在,则删除它并创建新表

csv_file_path = data_mapping_record.csv_file_path
original_csv_header = pandas.read_csv(csv_file_path).columns.tolist()
csv_data = pandas.read_csv(csv_file_path, skiprows=[0], names=original_csv_header, infer_datetime_format=True)
table_name = data_mapping_record.csv_file_path.split('/')[-1].split('.')[0]
engine = create_engine(
    'postgresql://username:password@localhost:5432/pandas_data')

# Delete table if already exits
engine.execute("""DROP TABLE IF EXISTS "%s" """ % (table_name))

# Write the pandas dataframe to database using sqlalchemy and pands.to_sql
csv_data_frame.to_sql(table_name, engine, chunksize=1000)

但我需要的是,如果没有删除表,如果表已经存在只是将数据附加到现有的数据,pandas to_sql方法有什么办法吗？

Answer 1

IIUC你可以简单地使用if_exists='append'参数:

csv_data_frame.to_sql(table_name, engine, if_exists='append', chunksize=1000)

来自docs:

if_exists:{'fail','replace','append'},默认'fail'

失败:如果表存在,则不执行任何操作.

replace:如果表存在,则删除它,重新创建它,然后插入数据.

append:如果表存在,则插入数据.创建如果不存在.