15

我有以下数据框

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)

但我需要的是,在不删除表的情况下,如果表已经存在,只需将数据附加到已经存在的表中,熊猫to_sql方法有什么办法吗?

4

1 回答 1

34

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

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

来自文档

if_exists : {'fail', 'replace', 'append'}, 默认'fail'</p>

失败:如果表存在,什么也不做。

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

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

于 2017-01-27T13:37:13.937 回答