Run a sample notebook by using Spark

Applies to: yesSQL Server 2019 (15.x)

This tutorial demonstrates how to load and run a notebook in Azure Data Studio on a SQL Server 2019 big data cluster. This allows data scientists and data engineers to run Python, R, or Scala code against the cluster.

Tip

If you prefer, you can download and run a script for the commands in this tutorial. For instructions, see the Spark samples on GitHub.

Prerequisites

Download the sample notebook file

Use the following instructions to load the sample notebook file spark-sql.ipynb into Azure Data Studio.

  1. Open a bash command prompt (Linux) or Windows PowerShell.

  2. Navigate to a directory where you want to download the sample notebook file to.

  3. Run the following curl command to download the notebook file from GitHub:

    curl https://raw.githubusercontent.com/Microsoft/sql-server-samples/master/samples/features/sql-big-data-cluster/spark/data-loading/transform-csv-files.ipynb -o transform-csv-files.ipynb
    

Open the notebook

The following steps show how to open the notebook file in Azure Data Studio:

  1. In Azure Data Studio, connect to the master instance of your big data cluster. For more information, see Connect to a big data cluster.

  2. Double-click on the HDFS/Spark gateway connection in the Servers window. Then select Open Notebook.

    Open notebook

  3. Wait for the Kernel and the target context (Attach to) to be populated. Set the Kernel to PySpark3, and set Attach to to the IP address of your big data cluster endpoint.

    Set Kernel and Attach to

Important

In Azure Data Studio, all Spark notebook types (Scala Spark, PySpark and SparkR) conventionally define some important Spark session related variables upon first cell execution. Those variables are: spark, sc, and sqlContext. When copying logic out of notebooks for batch submission (into a Python file to be run with azdata bdc spark batch create for example), make sure you define the variables accordingly.

Run the notebook cells

You can run each notebook cell by pressing the play button to the left of the cell. The results are shown in the notebook after the cell finishes running.

Run notebook cell

Run each of the cells in the sample notebook in succession. For more information about using notebooks with SQL Server Big Data Clusters, see the following resources:

Next steps

Learn more about notebooks: