Join us and get access to hundreds of tutorials and a community of expert Pythonistas.

Unlock This Lesson

This lesson is for members only. Join us and get access to hundreds of tutorials and a community of expert Pythonistas.

Unlock This Lesson

Hint: You can adjust the default video playback speed in your account settings.
Hint: You can set the default subtitles language in your account settings.
Sorry! Looks like there’s an issue with video playback 🙁 This might be due to a temporary outage or because of a configuration issue with your browser. Please see our video player troubleshooting guide to resolve the issue.

Working With SQL

00:00 Working with SQL files. pandas IO Tools can also read and write databases. In this next example, you’ll write your data to a database, but to get started, you’ll need to install the SQLAlchemy package, as seen onscreen now.

00:20 In addition, you’ll also need a database driver. Fortunately, Python has a built-in driver for SQLite databases, so that’s what we’ll be using. Once you have SQLAlchemy installed, import create_engine() and create a database engine as seen onscreen.

00:49 Now that you have all that set up, the next step is to create a DataFrame object as seen onscreen.

01:22 .astype() is a convenient method you can use to set multiple data types all at once using a dictionary as seen onscreen. Now the DataFrame can be saved to a database using the .to_sql() method as seen onscreen.

01:46 The parameter con is used to specify the database connection or engine that you want to use. The optional parameter index_label specifies how to call the database column with the row labels.

01:57 You’ll often see it take on the value 'ID'. This code had produced the file data.db, which can be read with an SQL file viewer.

02:08 The first column contains the row labels. The other columns correspond to the columns of the DataFrame. To omit writing row labels into the database, pass index=False to the .to_sql() method, and this will lead to the database seen onscreen now. There are a few more optional parameters, for example, you can use schema to specify the database schema and dtype to determine the types of the database columns. You can also use if_exists, which says what to do if a database with the same name and path already exists.

02:46 if_exists='fail' raises a ValueError and is the default. if_exists='replace' would drop the table and insert new values.

02:56 if_exists='append' inserts new values into the table. You can load the data from the database using read_sql().

03:15 The parameter index_col specifies the name of the column with the row labels. Note that this inserts an extra row after the header that starts with ID.

03:26 You can fix this behavior with the following line of code.

03:39 Now you have the same DataFrame object has before. Note that the content for Russia is now None instead of nan. If you want to fill in the missing values with nan, then you can use .fillna() as seen onscreen.

04:05 .fill_na() replaces all missing values with whatever you pass to value. Here, you passed float('nan'), which says to fill all missing values with nan.

04:16 Also note that you didn’t have to pass parse_dates=['IND_DAY'] to read_sql(). That’s because the database was able to detect that the last column contains dates. However, you could pass parse_dates if you’d like, but you’d still get the same results.

04:32 Next up, working with Python’s pickle file format.

KatMac on Sept. 23, 2021

Hi I have typed in the following code.....

from sqlalchemy import create_engine
import pandas as pd 

data   = pd.read_csv("FILES//data.csv")
engine = create_engine("sqlite:///data.db", echo=False)

db_types = {"POP": "float64", "AREA": "float64",
            "GDP": "float64", "IND_DAY": "datetime64"}

db_df = pd.DataFrame(data=data).T.astype(dtype=db_types)

and am receiving the following error:

KeyError: 'Only a column name can be used for the key in a dtype mappings argument.'


What have I done wrong? and Why did we not assign the COUNTRY type to text in db_types?

Bartosz Zaczyński RP Team on Sept. 24, 2021

@KatMac I think you want to transpose the data frame after having converted column types. Otherwise, the columns will have names corresponding to the consecutive numbers 0, 1, 2, and so on.

Try this instead:

db_df = pd.DataFrame(data=data).astype(dtype=db_types).T

KatMac on Sept. 24, 2021

Thank you Bartosz…

In my code, I have updated the db_df line as you suggested. The previous error has disappeared. When I run the code now the following displays (partial display to save space)

0     object
1     object
2     object
3     object
4     object
19    object
dtype: object

Become a Member to join the conversation.