Stata notebooks

Overview

Stata notebooks are available for those researchers who are more comfortable using Stata and its ecosystem. These are built off the same base image as python notebooks, but include the official pystata library to allow for the execution of Stata in a notebook environment.

Working with Stata in a notebook environment is slightly different than the Stata desktop application, in that we need to utilize python to pass data into Stata. This step is quite simple, and doesn't require any expertise in python – see working with tabular data below.

Enabling Stata notebooks

Because Stata is proprietary software, you will need to provide a license for Stata 16 or later in order to enable Stata notebooks on Redivis. Organizations can specify license information in their settings, which will make Stata notebooks available to all members of their organization. Alternatively, you can provide your own stata license in your workspace.

Base image and dependencies

Stata notebooks are based off the python notebook base image, and can combine both Stata and Python dependencies to create novel workflows.

To further customize your compute environment, you can specify various dependencies by clicking the Dependencies button at the top-right of your notebook. Here you will see three tabs: Packages, pre_install.sh, and post_install.sh.

Use packages to specify the python packages that you would like to install. When adding a new package, it will be pinned to the latest version of that package, but you can specify another version if preferred.

In order to install Stata packages via ssc, you should use the pre- and post- install shell scripts. These scripts are executed on either side of the python package installation, and are used to execute arbitrary code in the shell. Here you can execute stata code to run ssc install, and you can also use apt to install system packages (apt-get update && apt-get install -y <package>), or mamba to install from conda. E.g.

# A shell script. You can run Stata code here, 
# or install other system dependencies via apt-get

stata -e -q 'ssc install outreg2'

Working with tabular data

In order to load data into Stata, we first pull it into a data frame in python, and then pass that variable into Stata. If you're unfamiliar with python, you can just copy+paste the below into the first cell of your notebook to load the data in python.

# We first load the table via python, and then pass the dataframe into stata
import redivis
df = redivis.table("_source_").to_pandas_dataframe(dtype_backend="numpy")

View the Table.to_pandas_dataframe() python documentation ->

Next, in a separate cell, we use the %%stata "magic" at the start of our cell to specify that this is stata code. We include the -d df argument to pass in the df variable from python into Stata, and include the -force flag to tell Stata to overwrite any current dataset that we have.

%%stata -d df -force
/* Run stata code! All stata cells must be prefixed with %%stata */
describe()

Any subsequent cells that execute stata code should be prefixed by %%stata if they are more than one line, or by %stata if the code to be executed is all on one line:

%stata scatter mpg price

You can also use the %%mata command to execute Mata code:

%%mata
/* 
 Create the matrix X in Mata and then obtain its inverse, Xi. 
 Then, multiply Xi by the original matrix, X 
*/

X = (76, 53, 48 \ 53, 88, 46 \ 48, 46, 63)
Xi = invsym(X)
Xi
Xi*X

Working with geospatial data

Through various packages, Stata offers some support for geospatial datatypes. However, we can't pass geospatial data from python natively, and instead need to first create a shapefile that can then be loaded into Stata.

# This python code loads a geospatiale table and then writes it to a shapefile
geopandas_df = redivis.table("_source_").to_geopandas_dataframe(dtype_backend="numpy")
geopandas_df.to_file("out.shp")

View the Table.to_geopandas_dataframe() python documentation ->

%%stata
spshape2dta out.shp

Working with larger tables

If your data is too big to fit into memory, you may need to first download the data as a CSV, and then read that file into Stata:

import redivis
redivis.table("_source_").download("/scratch/table.csv", format="csv")

%%stata
import delimited "/scratch/table.csv"

Creating output tables

Redivis notebooks offer the ability to materialize notebook outputs as a new table node in your workflow. This table can then be processed by transforms, read into other notebooks, exported, or even re-imported into a dataset.

To create an output table, we first need to pass our Stata data back to python, using the -dout flag. We can then use the redivis.current_notebook().create_output_table() method in python to output our data.

If an output table for the notebook already exists, by default it will be overwritten. You can pass append=True to append, rather than overwrite, the table. In order for the append to succeed, all variables in the appended table, which are also present in the existing table, must have the same type.

%%stata -doutd df2
/*
  Once this cell executes, the current dataset will be pushed 
  to the python variable df2
*/
rename v* newv*

# Via python, pass this dataframe to the output table
# If append=True, subsequent calls will add to the existing table, 
#   rather than replacing it
redivis.current_notebook().create_output_table(df2, append=False)

Storing files

As you perform your analysis, you may generate files and figures that are stored on the notebook's hard disk. There are two locations that you should write files to: /out for persistent storage, and /scratch for temporary storage. By default, the output location is set to /scratch.

Any files written to persistent storage will be available when the notebook is stopped, and will be restored to the same state when the notebook is run again. Alternatively, any files written to temporary storage will only exist for the duration of the current notebook session.

%%stata
save "/out/my_dataset.dta"
outreg2 using /out/table.xls, replace