Skip to main content

Causal Methods Implemented in Python

Project description

CausaliPy

Causal Methods implemented in Python.

Installation

Install via

pip install causalipy

It might make sense to add the py-arrow dependency (which is currently required for the example):

pip install pyarrow

Example

To run a version of the multi-period difference-in-difference estimator as proposed by Callaway and Sant’Anna (2020) (this requires additionally pyarrow - e.g. via pip install pyarrow - to be installed currently):

from causalipy.did.multi_periods import MultiPeriodDid
import pandas as pd

url = "https://github.com/mohelm/causalipy-datasets/raw/main/mpdta-sample.feather"
data = pd.read_feather(url)

mpd_minimum_wage = MultiPeriodDid(
    data,
    outcome="lemp",
    treatment_indicator="treat",
    time_period_indicator="year",
    group_indiciator="first.treat",
    formula="~ 1",
)
mpd_minimum_wage.plot_treatment_effects()

This will give:

alt text

License

This project is licensed under the terms of the MIT license.

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

causalipy-0.1.1.tar.gz (7.9 kB view hashes)

Uploaded Source

Built Distribution

causalipy-0.1.1-py3-none-any.whl (8.8 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