Skip to main content

Koalas: pandas API on Apache Spark

Project description

pandas API on Apache Spark
Explore Koalas docs »

Live notebook · Issues · Mailing list
Help Thirsty Koalas Devasted by Recent Fires

The Koalas project makes data scientists more productive when interacting with big data, by implementing the pandas DataFrame API on top of Apache Spark.

pandas is the de facto standard (single-node) DataFrame implementation in Python, while Spark is the de facto standard for big data processing. With this package, you can:

  • Be immediately productive with Spark, with no learning curve, if you are already familiar with pandas.
  • Have a single codebase that works both with pandas (tests, smaller datasets) and with Spark (distributed datasets).

This project is currently in beta and is rapidly evolving, with a bi-weekly release cadence. We would love to have you try it and give us feedback, through our mailing lists or GitHub issues.

Try the Koalas 10 minutes tutorial on a live Jupyter notebook here. The initial launch can take up to several minutes.

Github Actions codecov Documentation Status Latest Release Conda Version Binder

Getting Started

The recommended way of installing Koalas is Conda as below.

conda install koalas -c conda-forge

You can use not only Conda but also multiple ways to install Koalas. See Installation for full instructions to install Koalas.

If you are a Databricks Runtime user, you can install Koalas using the Libraries tab on the cluster UI, or using dbutils in a notebook as below for the regular Databricks Runtime,

dbutils.library.installPyPI("koalas")
dbutils.library.restartPython()

or using conda with --no-deps option for Databricks Runtime for Machine Learning 6.0 and above, which provides all the required libraries.

%sh
conda install koalas -c conda-forge --no-deps

Note that Koalas requires Databricks Runtime 5.x or above. In the future, we will package Koalas out-of-the-box in both the regular Databricks Runtime and Databricks Runtime for Machine Learning.

Lastly, note that if your PyArrow version is 0.15+ and your PySpark version is lower than 3.0, it is best for you to set ARROW_PRE_0_15_IPC_FORMAT environment variable to 1 manually. Koalas will try its best to set it for you but it is impossible to set it if there is a Spark context already launched.

Now you can turn a pandas DataFrame into a Koalas DataFrame that is API-compliant with the former:

import databricks.koalas as ks
import pandas as pd

pdf = pd.DataFrame({'x':range(3), 'y':['a','b','b'], 'z':['a','b','b']})

# Create a Koalas DataFrame from pandas DataFrame
df = ks.from_pandas(pdf)

# Rename the columns
df.columns = ['x', 'y', 'z1']

# Do some operations in place:
df['x2'] = df.x * df.x

For more details, see Getting Started and Dependencies in the official documentation.

Contributing Guide

See Contributing Guide and Design Principles in the official documentation.

FAQ

See FAQ in the official documentation.

Best Practices

See Best Practices in the official documentation.

Koalas Talks and Blogs

See Koalas Talks and Blogs in the official documentation.

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

koalas-0.33.0.tar.gz (278.5 kB view details)

Uploaded Source

Built Distribution

koalas-0.33.0-py3-none-any.whl (964.3 kB view details)

Uploaded Python 3

File details

Details for the file koalas-0.33.0.tar.gz.

File metadata

  • Download URL: koalas-0.33.0.tar.gz
  • Upload date:
  • Size: 278.5 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/2.0.0 pkginfo/1.5.0.1 requests/2.22.0 setuptools/42.0.2.post20191203 requests-toolbelt/0.9.1 tqdm/4.40.2 CPython/3.7.5

File hashes

Hashes for koalas-0.33.0.tar.gz
Algorithm Hash digest
SHA256 ebe873314262f0fb9ea99738aa04dd1fb820f7ca02945ac189c945a2d0b89b57
MD5 44bb57a0e16c834261e6b20433d34de3
BLAKE2b-256 4a794d289a5aeeb76903eaf104be6f0a32ee270715a91c816f4756e0eebd3fd5

See more details on using hashes here.

Provenance

File details

Details for the file koalas-0.33.0-py3-none-any.whl.

File metadata

  • Download URL: koalas-0.33.0-py3-none-any.whl
  • Upload date:
  • Size: 964.3 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/2.0.0 pkginfo/1.5.0.1 requests/2.22.0 setuptools/42.0.2.post20191203 requests-toolbelt/0.9.1 tqdm/4.40.2 CPython/3.7.5

File hashes

Hashes for koalas-0.33.0-py3-none-any.whl
Algorithm Hash digest
SHA256 1e39849e5d211150e0a1c2c5def1222babfc33d5db4d6508d6d4b5b7f203f8c7
MD5 cb4990d35ff4690134f7e8dfdeebae58
BLAKE2b-256 85e3e1a24f6e8d110ef03bf4023b75581a42583643b7f38a5245a086175b4053

See more details on using hashes here.

Provenance

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