No module named 'databricks' after installing koalas

[loungehub]

I have installed koalas using conda install.
However when I try the following import, I get an ModuleNotFoundError.
Can you help me to solve this issue?

import databricks.koalas as ks
Traceback (most recent call last):
File "", line 1, in
ModuleNotFoundError: No module named 'databricks'

[itholic]

Can you use Pandas API on Spark instead of Koalas as the Koalas was ported into PySpark and it is no longer maintained??

You can just simply install pyspark and change your import as below then your existing Koalas code just works:

# import databricks.koalas as ks
import pyspark.pandas as ks

(your existing Koalas code)

[mishuhaque]

+1 to @itholic's suggestion — I switched from Koalas to pyspark.pandas recently, and the transition was smooth.
Just a heads-up: if you were using a conda environment, make sure pyspark is installed in that same env:

conda install pyspark

or use pip install pyspark if you're mixing pip-based installs.

Also, databricks.koalas is now deprecated, so future updates will only happen in PySpark's pandas API.