Skip to main content

Dragon Imputation

A Polars plugin for performing imputation on missing values in DataFrames.

Works on both continuous and categorical variables in numerical form. Categorical variables should be represented as integers (e.g., 0, 1, 2, ...).

MICE imputation uses random decision tree models to predict missing values based on the observed values in the DataFrame. The algorithm iteratively fills in missing values by modeling each variable with missing data as a function of other variables in a round-robin fashion.

Installation

pip install dragon-imputation
uv add dragon-imputation

Usage

import polars as pl
from dragon_imputation import mice_impute

# Create a sample DataFrame with missing values
df = pl.DataFrame({
    "A": [1.1, 2.2, None, 4.4],
    "B_cat": [None, 2, 3, 4],
    "C": [1.0, None, 3.0, 4.0]
})

# Perform MICE imputation with 5 iterations
imputed_df = mice_impute(df=df, cat_cols=[1], max_iterations=5)

Arguments

mice_impute() function takes the following arguments:

  • df: A Polars DataFrame containing missing values.
  • cat_cols: An optional sequence of column indices that are categorical. If None, all columns are treated as continuous.
  • max_iterations: The maximum number of iterations for the MICE algorithm (default is 30).
  • max_depth: The maximum depth for the decision tree model used in imputation (default is None for no depth limit).
  • verbose: The verbosity level for logging (default is 2).
  • seed: The random seed for reproducibility (default is 42).
  • n_jobs: The number of threads to use for parallel processing (default is None, which uses all available CPU cores).

Metadata

Release files for dragon-imputation 0.2.1

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Source distribution (sdist)

Source distribution for dragon-imputation 0.2.1
File Size Uploaded Provenance Actions
dragon_imputation-0.2.1.tar.gz 35.5 kB Without provenance Details for dragon_imputation-0.2.1.tar.gz

Built distributions (wheels)

Table of built distributions (wheels) for dragon-imputation 0.2.1
File Actions
dragon_imputation-0.2.1-cp38-abi3-win_arm64.whl CPython 3.8 abi3 Windows ARM64 Details for dragon_imputation-0.2.1-cp38-abi3-win_arm64.whl
dragon_imputation-0.2.1-cp38-abi3-win_amd64.whl CPython 3.8 abi3 Windows x86-64 Details for dragon_imputation-0.2.1-cp38-abi3-win_amd64.whl
dragon_imputation-0.2.1-cp38-abi3-win32.whl CPython 3.8 abi3 Windows x86-32 Details for dragon_imputation-0.2.1-cp38-abi3-win32.whl
dragon_imputation-0.2.1-cp38-abi3-musllinux_1_2_x86_64.whl CPython 3.8 abi3 Linux musl 1.2+ x86-64 Details for dragon_imputation-0.2.1-cp38-abi3-musllinux_1_2_x86_64.whl
dragon_imputation-0.2.1-cp38-abi3-musllinux_1_2_i686.whl CPython 3.8 abi3 Linux musl 1.2+ x86-32 Details for dragon_imputation-0.2.1-cp38-abi3-musllinux_1_2_i686.whl
dragon_imputation-0.2.1-cp38-abi3-musllinux_1_2_armv7l.whl CPython 3.8 abi3 Linux musl 1.2+ ARMv7l Details for dragon_imputation-0.2.1-cp38-abi3-musllinux_1_2_armv7l.whl
dragon_imputation-0.2.1-cp38-abi3-musllinux_1_2_aarch64.whl CPython 3.8 abi3 Linux musl 1.2+ ARM64 Details for dragon_imputation-0.2.1-cp38-abi3-musllinux_1_2_aarch64.whl
dragon_imputation-0.2.1-cp38-abi3-manylinux_2_17_x86_64.manylinux2014_x86_64.whl CPython 3.8 abi3 Linux glibc 2.17+ x86-64 Details for dragon_imputation-0.2.1-cp38-abi3-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
dragon_imputation-0.2.1-cp38-abi3-manylinux_2_17_i686.manylinux2014_i686.whl CPython 3.8 abi3 Linux glibc 2.17+ x86-32 Details for dragon_imputation-0.2.1-cp38-abi3-manylinux_2_17_i686.manylinux2014_i686.whl
dragon_imputation-0.2.1-cp38-abi3-manylinux_2_17_armv7l.manylinux2014_armv7l.whl CPython 3.8 abi3 Linux glibc 2.17+ ARMv7l Details for dragon_imputation-0.2.1-cp38-abi3-manylinux_2_17_armv7l.manylinux2014_armv7l.whl
dragon_imputation-0.2.1-cp38-abi3-manylinux_2_17_aarch64.manylinux2014_aarch64.whl CPython 3.8 abi3 Linux glibc 2.17+ ARM64 Details for dragon_imputation-0.2.1-cp38-abi3-manylinux_2_17_aarch64.manylinux2014_aarch64.whl
dragon_imputation-0.2.1-cp38-abi3-macosx_11_0_arm64.whl CPython 3.8 abi3 macOS 11.0+ ARM64 Details for dragon_imputation-0.2.1-cp38-abi3-macosx_11_0_arm64.whl
dragon_imputation-0.2.1-cp38-abi3-macosx_10_12_x86_64.whl CPython 3.8 abi3 macOS 10.12+ x86-64 Details for dragon_imputation-0.2.1-cp38-abi3-macosx_10_12_x86_64.whl

Total release size: 71.5 MB

Release files / dragon_imputation-0.2.1.tar.gz

Download URL dragon_imputation-0.2.1.tar.gz
Size 35.5 kB
Tags Source
SHA-256 checksum
How to use checksums
445da773e17209753a01a04391d9123dd877d1ed084bdcbf9ba63a984c5bb3d4
BLAKE2b-256 checksum
How to use checksums
6f55e31d31d272ef1d58a19554aae07c1ff9a5ca10e10c39a0a1be9a93fb3c8e
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
Yes
Uploaded via uv/0.13.1 {"installer":{"name":"uv","version":"0.13.1","subcommand":["publish"]},"python":null,"implementation":{"name":null,"version":null},"distro":{"name":"Ubuntu","version":"24.04","id":"noble","libc":null},"system":{"name":null,"release":null},"cpu":null,"openssl_version":null,"setuptools_version":null,"rustc_version":null,"ci":true}

Release files / dragon_imputation-0.2.1-cp38-abi3-win_arm64.whl

Download URL dragon_imputation-0.2.1-cp38-abi3-win_arm64.whl
Size 4.5 MB
Tags CPython 3.8 Windows ARM64 abi3
SHA-256 checksum
How to use checksums
c5f31823e1fbf365ce0ebfb5d3e72e8e99979a899cfcbe5e93ba40fd76424e59
BLAKE2b-256 checksum
How to use checksums
31fd7019ecdb1b80f6c8a254af4ede9709292c26f5d6ab75afe5543c48b7e6e4
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
Yes
Uploaded via uv/0.13.1 {"installer":{"name":"uv","version":"0.13.1","subcommand":["publish"]},"python":null,"implementation":{"name":null,"version":null},"distro":{"name":"Ubuntu","version":"24.04","id":"noble","libc":null},"system":{"name":null,"release":null},"cpu":null,"openssl_version":null,"setuptools_version":null,"rustc_version":null,"ci":true}

Release files / dragon_imputation-0.2.1-cp38-abi3-win_amd64.whl

Download URL dragon_imputation-0.2.1-cp38-abi3-win_amd64.whl
Size 5.0 MB
Tags CPython 3.8 Windows x86-64 abi3
SHA-256 checksum
How to use checksums
ec8bd73d8b3151327c6f4d5572d28f33f1279bd1d14d80b65cefb6d38bbddc14
BLAKE2b-256 checksum
How to use checksums
8cb880d7194ed314cb99a656bef3a895468752ca3ca073998bc492a8b9198c46
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
Yes
Uploaded via uv/0.13.1 {"installer":{"name":"uv","version":"0.13.1","subcommand":["publish"]},"python":null,"implementation":{"name":null,"version":null},"distro":{"name":"Ubuntu","version":"24.04","id":"noble","libc":null},"system":{"name":null,"release":null},"cpu":null,"openssl_version":null,"setuptools_version":null,"rustc_version":null,"ci":true}

Release files / dragon_imputation-0.2.1-cp38-abi3-win32.whl

Download URL dragon_imputation-0.2.1-cp38-abi3-win32.whl
Size 4.3 MB
Tags CPython 3.8 Windows x86-32 abi3
SHA-256 checksum
How to use checksums
5c8ab38e6ab87758cba277220fee8dac05bd6e9f63e286f0a84a9c8e1a64fb0d
BLAKE2b-256 checksum
How to use checksums
404a5c77c5b0ba5a859cbde9de5efe989b2c7e8982b50123d26e29ebccf79cb5
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
Yes
Uploaded via uv/0.13.1 {"installer":{"name":"uv","version":"0.13.1","subcommand":["publish"]},"python":null,"implementation":{"name":null,"version":null},"distro":{"name":"Ubuntu","version":"24.04","id":"noble","libc":null},"system":{"name":null,"release":null},"cpu":null,"openssl_version":null,"setuptools_version":null,"rustc_version":null,"ci":true}

Release files / dragon_imputation-0.2.1-cp38-abi3-musllinux_1_2_x86_64.whl

Download URL dragon_imputation-0.2.1-cp38-abi3-musllinux_1_2_x86_64.whl
Size 6.2 MB
Tags CPython 3.8 Linux musl 1.2+ x86-64 abi3
SHA-256 checksum
How to use checksums
09b8a56a9116c5a6a05c4f6749d284b6405404c3e734ccb4e0e9a688cf001a83
BLAKE2b-256 checksum
How to use checksums
fe07b59dc5e044a029ac3b705a0e811e4c880dea5651f3498dcfedfb3645657d
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
Yes
Uploaded via uv/0.13.1 {"installer":{"name":"uv","version":"0.13.1","subcommand":["publish"]},"python":null,"implementation":{"name":null,"version":null},"distro":{"name":"Ubuntu","version":"24.04","id":"noble","libc":null},"system":{"name":null,"release":null},"cpu":null,"openssl_version":null,"setuptools_version":null,"rustc_version":null,"ci":true}

Release files / dragon_imputation-0.2.1-cp38-abi3-musllinux_1_2_i686.whl

Download URL dragon_imputation-0.2.1-cp38-abi3-musllinux_1_2_i686.whl
Size 6.5 MB
Tags CPython 3.8 Linux musl 1.2+ x86-32 abi3
SHA-256 checksum
How to use checksums
4694c1564b123d4531a1cf59ef4e1458bfd7b47ab8cc2885dc41c0e946f621c7
BLAKE2b-256 checksum
How to use checksums
c5a60f4aad2ca2990f67f68adf0db4771fb56db53bcfdd41e8ceb32ac158673b
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
Yes
Uploaded via uv/0.13.1 {"installer":{"name":"uv","version":"0.13.1","subcommand":["publish"]},"python":null,"implementation":{"name":null,"version":null},"distro":{"name":"Ubuntu","version":"24.04","id":"noble","libc":null},"system":{"name":null,"release":null},"cpu":null,"openssl_version":null,"setuptools_version":null,"rustc_version":null,"ci":true}

Release files / dragon_imputation-0.2.1-cp38-abi3-musllinux_1_2_armv7l.whl

Download URL dragon_imputation-0.2.1-cp38-abi3-musllinux_1_2_armv7l.whl
Size 6.1 MB
Tags CPython 3.8 Linux musl 1.2+ ARMv7l abi3
SHA-256 checksum
How to use checksums
fc165e7254f0c7a5b2b88a7eae860c61b8d54fb60d5bee48d4ca3c76ecf35927
BLAKE2b-256 checksum
How to use checksums
69a4c4f80d0e100e379f5f2416c46da89b782e304475b9034e5886c9e2477c9b
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
Yes
Uploaded via uv/0.13.1 {"installer":{"name":"uv","version":"0.13.1","subcommand":["publish"]},"python":null,"implementation":{"name":null,"version":null},"distro":{"name":"Ubuntu","version":"24.04","id":"noble","libc":null},"system":{"name":null,"release":null},"cpu":null,"openssl_version":null,"setuptools_version":null,"rustc_version":null,"ci":true}

Release files / dragon_imputation-0.2.1-cp38-abi3-musllinux_1_2_aarch64.whl

Download URL dragon_imputation-0.2.1-cp38-abi3-musllinux_1_2_aarch64.whl
Size 5.8 MB
Tags CPython 3.8 Linux musl 1.2+ ARM64 abi3
SHA-256 checksum
How to use checksums
0952323d46e21e83986b830ec96306574987ed707f8ef3b4e94468e58586118a
BLAKE2b-256 checksum
How to use checksums
624be442d48a474fa688ba8c7ac4cf8685745679d03de6654527064faa8f76d5
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
Yes
Uploaded via uv/0.13.1 {"installer":{"name":"uv","version":"0.13.1","subcommand":["publish"]},"python":null,"implementation":{"name":null,"version":null},"distro":{"name":"Ubuntu","version":"24.04","id":"noble","libc":null},"system":{"name":null,"release":null},"cpu":null,"openssl_version":null,"setuptools_version":null,"rustc_version":null,"ci":true}

Release files / dragon_imputation-0.2.1-cp38-abi3-manylinux_2_17_x86_64.manylinux2014_x86_64.whl

Download URL dragon_imputation-0.2.1-cp38-abi3-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
Size 5.4 MB
Tags CPython 3.8 Linux glibc 2.17+ x86-64 abi3
SHA-256 checksum
How to use checksums
4c135b77caf98da28e6c8937f41c6333758aa1be8a532e8048e95ca06bd9e558
BLAKE2b-256 checksum
How to use checksums
df2221035842c8c6d8644b6fce39e33007028b17c2fd4af5ebaf051d85d6ff01
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
Yes
Uploaded via uv/0.13.1 {"installer":{"name":"uv","version":"0.13.1","subcommand":["publish"]},"python":null,"implementation":{"name":null,"version":null},"distro":{"name":"Ubuntu","version":"24.04","id":"noble","libc":null},"system":{"name":null,"release":null},"cpu":null,"openssl_version":null,"setuptools_version":null,"rustc_version":null,"ci":true}

Release files / dragon_imputation-0.2.1-cp38-abi3-manylinux_2_17_i686.manylinux2014_i686.whl

Download URL dragon_imputation-0.2.1-cp38-abi3-manylinux_2_17_i686.manylinux2014_i686.whl
Size 6.5 MB
Tags CPython 3.8 Linux glibc 2.17+ x86-32 abi3
SHA-256 checksum
How to use checksums
29b871cc01f8c7af1a86bb10a5ee5db46f153101df59e2e13c8ffccaf2bf5d5a
BLAKE2b-256 checksum
How to use checksums
0b201d6b4c753b79749bd5dc42666ab4ec3cbf904afd055a1f1cb67a8185c790
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
Yes
Uploaded via uv/0.13.1 {"installer":{"name":"uv","version":"0.13.1","subcommand":["publish"]},"python":null,"implementation":{"name":null,"version":null},"distro":{"name":"Ubuntu","version":"24.04","id":"noble","libc":null},"system":{"name":null,"release":null},"cpu":null,"openssl_version":null,"setuptools_version":null,"rustc_version":null,"ci":true}

Release files / dragon_imputation-0.2.1-cp38-abi3-manylinux_2_17_armv7l.manylinux2014_armv7l.whl

Download URL dragon_imputation-0.2.1-cp38-abi3-manylinux_2_17_armv7l.manylinux2014_armv7l.whl
Size 5.8 MB
Tags CPython 3.8 Linux glibc 2.17+ ARMv7l abi3
SHA-256 checksum
How to use checksums
ca8efbe8f151e4b28356582fc9f1237bbc810b58e47798060c5ffbc72ca10e1e
BLAKE2b-256 checksum
How to use checksums
1529af8663771151610e9da81ef4cb5137f18374a0b43f74bbcc25c4cc87b9d8
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
Yes
Uploaded via uv/0.13.1 {"installer":{"name":"uv","version":"0.13.1","subcommand":["publish"]},"python":null,"implementation":{"name":null,"version":null},"distro":{"name":"Ubuntu","version":"24.04","id":"noble","libc":null},"system":{"name":null,"release":null},"cpu":null,"openssl_version":null,"setuptools_version":null,"rustc_version":null,"ci":true}

Release files / dragon_imputation-0.2.1-cp38-abi3-manylinux_2_17_aarch64.manylinux2014_aarch64.whl

Download URL dragon_imputation-0.2.1-cp38-abi3-manylinux_2_17_aarch64.manylinux2014_aarch64.whl
Size 5.7 MB
Tags CPython 3.8 Linux glibc 2.17+ ARM64 abi3
SHA-256 checksum
How to use checksums
2a15020c7edc9052e0d8036bd4b3e0adef8ad8d2305d4d681a41b1e6e347730d
BLAKE2b-256 checksum
How to use checksums
e1ede179f4dfe38cb2db2ab6fe55a33fdc97b6fb661f92908fb5dca3d5578f4d
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
Yes
Uploaded via uv/0.13.1 {"installer":{"name":"uv","version":"0.13.1","subcommand":["publish"]},"python":null,"implementation":{"name":null,"version":null},"distro":{"name":"Ubuntu","version":"24.04","id":"noble","libc":null},"system":{"name":null,"release":null},"cpu":null,"openssl_version":null,"setuptools_version":null,"rustc_version":null,"ci":true}

Release files / dragon_imputation-0.2.1-cp38-abi3-macosx_11_0_arm64.whl

Download URL dragon_imputation-0.2.1-cp38-abi3-macosx_11_0_arm64.whl
Size 4.7 MB
Tags CPython 3.8 abi3 macOS 11.0+ ARM64
SHA-256 checksum
How to use checksums
1d78673e4edee96c35ea3ecbef9e47c84ce73f1831f3f2d06e23d1ad76482072
BLAKE2b-256 checksum
How to use checksums
d593c278cc23003cc8df91c3e4353e9954d4529cd98bb247d0b29b272f272ad1
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
Yes
Uploaded via uv/0.13.1 {"installer":{"name":"uv","version":"0.13.1","subcommand":["publish"]},"python":null,"implementation":{"name":null,"version":null},"distro":{"name":"Ubuntu","version":"24.04","id":"noble","libc":null},"system":{"name":null,"release":null},"cpu":null,"openssl_version":null,"setuptools_version":null,"rustc_version":null,"ci":true}

Release files / dragon_imputation-0.2.1-cp38-abi3-macosx_10_12_x86_64.whl

Download URL dragon_imputation-0.2.1-cp38-abi3-macosx_10_12_x86_64.whl
Size 4.9 MB
Tags CPython 3.8 abi3 macOS 10.12+ x86-64
SHA-256 checksum
How to use checksums
fd06e3564355d6bfa406da679204443e115a983096a88e1fe5ce3d0622b761fa
BLAKE2b-256 checksum
How to use checksums
c20872d134c7ed7eeef1be72561d8dd1f9039b66d29b745a994dfa1ea4589bbc
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
Yes
Uploaded via uv/0.13.1 {"installer":{"name":"uv","version":"0.13.1","subcommand":["publish"]},"python":null,"implementation":{"name":null,"version":null},"distro":{"name":"Ubuntu","version":"24.04","id":"noble","libc":null},"system":{"name":null,"release":null},"cpu":null,"openssl_version":null,"setuptools_version":null,"rustc_version":null,"ci":true}

Security for dragon-imputation 0.2.1

Provenance

Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.

No provenance found

None of the files in this release were uploaded with provenance metadata.

A note on package security

PyPI is a repository for distributing packages, not a service for auditing them. While third-party security researchers and our staff do their best to detect and remove malicious packages, PyPI cannot guarantee the quality, security or suitability of any given package.

To learn more about protecting yourself from malicious actors, visit our package security documentation.

Anthropic, PBC Visionary sponsor Bloomberg Visionary sponsor Hudson River Trading Visionary sponsor Meta Visionary sponsor NVIDIA Visionary sponsor Microsoft Sustainability sponsor Depot Continuous Integration AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page