pandas_sql_test.py

import pandas as pd
import sqlalchemy
from sklearn import datasets

# create iris dataframe
iris = datasets.load_iris()
iris_data = pd.DataFrame(iris.data)
iris_data.columns = ["sep_length", "sepal_width","petal_length","petal_width" ]
iris_data["species"] = map(lambda i:iris.target_names[i], iris.target)

#insert into sqlite
engine = sqlalchemy.create_engine("sqlite:///test_db")
iris_data.to_sql("iris_dat", engine, index=False, if_exists="replace")

# insert into postgresql
engine = sqlalchemy.create_engine("postgresql://<user>:<passwd>@localhost/test_db")
iris_data.to_sql("iris_dat", engine, index=False, if_exists="replace")