Skip to main content

A Pythonic Package for Regression Discontinuity

Project description

State of the Art RD Packages For Python

Installation

With pip for python: pip install py_rdpackages

Important: you need to have the original rdrobust installed in R: install.packages('rdrobust')

Notes:

  • rdrobust requires the newest version of ggplot2 so please: install.packages('ggplot2')
  • The rdrobust and ggplot2 versions in Anaconda lag behind the current release for R. Please install in R directly as opposed to through conda.

Introduction and Use

These packages are a work in progress, but are an attempt to create a wrapper to implement the wonderful RD packages found here (https://sites.google.com/site/rdpackages/rdrobust) which utilize R or Stata, so that they can be used in Python directly.

There are three packages in py_rdpackages:

  1. rdplot creates plots of the regression discontinuity with a variety of options.
  2. rdrobust does the RD and reports the regression results.
  3. rdbwselect selects the optimal bandwidth size.

Requirements and Stability

Use of the programs requires all of the following packages in Python:

  1. rpy2 for running R in Python
  2. matplotlib and seaborn for producing high quality graphics
  3. pandas, numpy and for data manipulation and dataframe reading

Currently tested and stable for:

  • rdrobust version 0.99.7
  • ggplot2 version 3.3.0

Limitations

  1. ryp2 produced slow pandas DF to R DF conversions, so I use pd.df.to_csv('temp_file_for_rd.csv') as a solution and then delete that same file after doing the analysis. This should be fixed in future versions.
  2. Of course, Python calling R and then converting back to python is not ideal. Some future version should code this from scratch.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

py_rdpackages-1.0.1.tar.gz (11.4 kB view hashes)

Uploaded Source

Built Distribution

py_rdpackages-1.0.1-py3-none-any.whl (22.9 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page