Redigera

Dela via


Python plugin

Applies to: ✅ Microsoft FabricAzure Data Explorer

The Python plugin runs a user-defined function (UDF) using a Python script. The Python script gets tabular data as its input, and produces tabular output. The plugin's runtime is hosted in sandboxes, running on the cluster's nodes.

Syntax

T | evaluate [hint.distribution = (single | per_node)] [hint.remote = (auto | local)] python(output_schema, script [, script_parameters] [, external_artifacts][, spill_to_disk])

Learn more about syntax conventions.

Parameters

Name Type Required Description
output_schema string ✔️ A type literal that defines the output schema of the tabular data, returned by the Python code. The format is: typeof(ColumnName: ColumnType[, ...]). For example, typeof(col1:string, col2:long). To extend the input schema, use the following syntax: typeof(*, col1:string, col2:long).
script string ✔️ The valid Python script to execute. To generate multi-line strings, see Usage tips.
script_parameters dynamic A property bag of name value pairs to be passed to the Python script as the reserved kargs dictionary. For more information, see Reserved Python variables.
hint.distribution string A hint for the plugin's execution to be distributed across multiple cluster nodes. The default value is single. single means a single instance of the script will run over the entire query data. per_node means that if the query before the Python block is distributed, an instance of the script will run on each node, on the data that it contains.
hint.remote string This hint is only relevant for cross cluster queries. The default value is auto. auto means the server decides automatically in which cluster the Python code is executed. Setting the value to local forces executing the Python code on the local cluster. Use it in case the Python plugin is disabled on the remote cluster.
external_artifacts dynamic A property bag of name and URL pairs for artifacts that are accessible from cloud storage. See more in Using external artifacts.
spill_to_disk bool Specifies an alternative method for serializing the input table to the Python sandbox. For serializing big tables set it to true to speed up the serialization and significantly reduce the sandbox memory consumption. Default is true.

Reserved Python variables

The following variables are reserved for interaction between Kusto Query Language and the Python code.

  • df: The input tabular data (the values of T above), as a pandas DataFrame.
  • kargs: The value of the script_parameters argument, as a Python dictionary.
  • result: A pandas DataFrame created by the Python script, whose value becomes the tabular data that gets sent to the Kusto query operator that follows the plugin.

Enable the plugin

The plugin is disabled by default. Before you start, review the list of prerequisites. To enable the plugin and select the version of the Python image, see Enable language extensions on your cluster.

Python sandbox image

To change the version of the Python image to a different managed image or a custom image, see Change the Python language extensions image on your cluster.

To see the list of packages for the different Python images, see Python package reference.

Note

  • By default, the plugin imports numpy as np and pandas as pd. Optionally, you can import other modules as needed.
  • Some packages might be incompatible with the limitations enforced by the sandbox where the plugin is run.

Use ingestion from query and update policy

  • Use the plugin in queries that are:
    • Defined as part of an update policy, whose source table is ingested to using non-streaming ingestion.
    • Run as part of a command that ingests from a query, such as .set-or-append.
  • You can't use the plugin in a query that is defined as part of an update policy, whose source table is ingested using streaming ingestion.

Examples

range x from 1 to 360 step 1
| evaluate python(
//
typeof(*, fx:double),               //  Output schema: append a new fx column to original table 
```
result = df
n = df.shape[0]
g = kargs["gain"]
f = kargs["cycles"]
result["fx"] = g * np.sin(df["x"]/n*2*np.pi*f)
```
, bag_pack('gain', 100, 'cycles', 4)    //  dictionary of parameters
)
| render linechart 

Screenshot of sine demo showing query result.

print "This is an example for using 'external_artifacts'"
| evaluate python(
    typeof(File:string, Size:string), ```if 1:
    import os
    result = pd.DataFrame(columns=['File','Size'])
    sizes = []
    path = '.\\\\Temp'
    files = os.listdir(path)
    result['File']=files
    for file in files:
        sizes.append(os.path.getsize(path + '\\\\' + file))
    result['Size'] = sizes
    ```,
    external_artifacts = 
        dynamic({"this_is_my_first_file":"https://kustoscriptsamples.blob.core.windows.net/samples/R/sample_script.r",
                 "this_is_a_script":"https://kustoscriptsamples.blob.core.windows.net/samples/python/sample_script.py"})
)
File Size
this_is_a_script 120
this_is_my_first_file 105

Performance tips

  • Reduce the plugin's input dataset to the minimum amount required (columns/rows).
    • Use filters on the source dataset, when possible, with Kusto's query language.
    • To do a calculation on a subset of the source columns, project only those columns before invoking the plugin.
  • Use hint.distribution = per_node whenever the logic in your script is distributable.
  • Use Kusto's query language whenever possible, to implement the logic of your Python script.

Usage tips

  • To generate multi-line strings containing the Python script in your query editor, copy your Python script from your favorite Python editor (Jupyter, Visual Studio Code, PyCharm, and so on), paste it in your query editor, and then enclose the full script between lines containing three consecutive backticks. For example:

    ```
    python code
    ```

  • Use the externaldata operator to obtain the content of a script that you've stored in an external location, such as Azure Blob storage.

Example

    let script = 
        externaldata(script:string)
        [h'https://kustoscriptsamples.blob.core.windows.net/samples/python/sample_script.py']
        with(format = raw);
    range x from 1 to 360 step 1
    | evaluate python(
        typeof(*, fx:double),
        toscalar(script), 
        bag_pack('gain', 100, 'cycles', 4))
    | render linechart 

Using External Artifacts

External artifacts from cloud storage can be made available for the script and used at runtime.

The URLs referenced by the external artifacts property must be:

Note

When authenticating external artifacts using Managed Identities, the SandboxArtifacts usage must be defined on the cluster level managed identity policy.

The artifacts are made available for the script to consume from a local temporary directory, .\Temp. The names provided in the property bag are used as the local file names. See Examples.

For information regarding referencing external packages, see Install packages for the Python plugin.

Refreshing external artifact cache

External artifact files utilized in queries are cached on your cluster. If you make updates to your files in cloud storage and require immediate synchronization with your cluster, you can use the .clear cluster cache external-artifacts command. This command clears the cached files and ensures that subsequent queries run with the latest version of the artifacts.

Install packages for the Python plugin

In most use cases, you might prefer to create a custom image.

You might want to install package(s) yourself, for the following reasons:

  • You don't have permissions to create a custom image.
  • The package is private.
  • You prefer to create an ad hoc package installation for testing and don't want the overhead of creating a custom image.

Install packages as follows:

Prerequisites

  1. Create a blob container to host the packages, preferably in the same place as your cluster. For example, https://artifactswestus.blob.core.windows.net/python, assuming your cluster is in West US.

  2. Alter the cluster's callout policy to allow access to that location.

    • This change requires AllDatabasesAdmin permissions.

    • For example, to enable access to a blob located in https://artifactswestus.blob.core.windows.net/python, run the following command:

    .alter-merge cluster policy callout @'[ { "CalloutType": "sandbox_artifacts", "CalloutUriRegex": "artifactswestus\\.blob\\.core\\.windows\\.net/python/","CanCall": true } ]'
    

Install packages

  1. For public packages in PyPi or other channels, download the package and its dependencies.

    • From a cmd window in your local Windows Python environment, run:
    pip wheel [-w download-dir] package-name.
    
  2. Create a ZIP file that contains the required package and its dependencies.

    • For private packages, zip the folder of the package and the folders of its dependencies.
    • For public packages, zip the files that were downloaded in the previous step.

    Note

    • Make sure to download the package that is compatible to the Python engine and the platform of the sandbox runtime (currently 3.6.5 on Windows)
    • Make sure to zip the .whl files themselves, and not their parent folder.
    • You can skip .whl files for packages that already exist with the same version in the base sandbox image.
  3. Upload the zipped file to a blob in the artifacts location (from step 1).

  4. Call the python plugin.

    • Specify the external_artifacts parameter with a property bag of name and reference to the ZIP file (the blob's URL, including a SAS token).
    • In your inline python code, import Zipackage from sandbox_utils and call its install() method with the name of the ZIP file.

Example

Install the Faker package that generates fake data.

range ID from 1 to 3 step 1 
| extend Name=''
| evaluate python(typeof(*), ```if 1:
    from sandbox_utils import Zipackage
    Zipackage.install("Faker.zip")
    from faker import Faker
    fake = Faker()
    result = df
    for i in range(df.shape[0]):
        result.loc[i, "Name"] = fake.name()
    ```,
    external_artifacts=bag_pack('faker.zip', 'https://artifacts.blob.core.windows.net/Faker.zip?*** REPLACE WITH YOUR SAS TOKEN ***'))
ID Name
1 Gary Tapia
2 Emma Evans
3 Ashley Bowen

For more examples of UDF functions that use the Python plugin, see the Functions library.

The Python plugin runs a user-defined function (UDF) using a Python script. The Python script gets tabular data as its input, and produces tabular output.

Syntax

T | evaluate [hint.distribution = (single | per_node)] [hint.remote = (auto | local)] python(output_schema, script [, script_parameters] [, spill_to_disk])

Learn more about syntax conventions.

Parameters

Name Type Required Description
output_schema string ✔️ A type literal that defines the output schema of the tabular data, returned by the Python code. The format is: typeof(ColumnName: ColumnType[, ...]). For example, typeof(col1:string, col2:long). To extend the input schema, use the following syntax: typeof(*, col1:string, col2:long).
script string ✔️ The valid Python script to execute. To generate multi-line strings, see Usage tips.
script_parameters dynamic A property bag of name value pairs to be passed to the Python script as the reserved kargs dictionary. For more information, see Reserved Python variables.
hint.distribution string A hint for the plugin's execution to be distributed across multiple cluster nodes. The default value is single. single means a single instance of the script will run over the entire query data. per_node means that if the query before the Python block is distributed, an instance of the script will run on each node, on the data that it contains.
hint.remote string This hint is only relevant for cross cluster queries. The default value is auto. auto means the server decides automatically in which cluster the Python code is executed. Setting the value to local forces executing the Python code on the local cluster. Use it in case the Python plugin is disabled on the remote cluster.
spill_to_disk bool Specifies an alternative method for serializing the input table to the Python sandbox. For serializing big tables set it to true to speed up the serialization and significantly reduce the sandbox memory consumption. Default is true.

Reserved Python variables

The following variables are reserved for interaction between Kusto Query Language and the Python code.

  • df: The input tabular data (the values of T above), as a pandas DataFrame.
  • kargs: The value of the script_parameters argument, as a Python dictionary.
  • result: A pandas DataFrame created by the Python script, whose value becomes the tabular data that gets sent to the Kusto query operator that follows the plugin.

Enable the plugin

The plugin is disabled by default. Before you start, enable the Python plugin in your KQL database.

Python sandbox image

To see the list of packages for the different Python images, see Python package reference.

Note

  • By default, the plugin imports numpy as np and pandas as pd. Optionally, you can import other modules as needed.
  • Some packages might be incompatible with the limitations enforced by the sandbox where the plugin is run.

Use ingestion from query and update policy

  • Use the plugin in queries that are:
    • Defined as part of an update policy, whose source table is ingested to using non-streaming ingestion.
    • Run as part of a command that ingests from a query, such as .set-or-append.
  • You can't use the plugin in a query that is defined as part of an update policy, whose source table is ingested using streaming ingestion.

Examples

range x from 1 to 360 step 1
| evaluate python(
//
typeof(*, fx:double),               //  Output schema: append a new fx column to original table 
```
result = df
n = df.shape[0]
g = kargs["gain"]
f = kargs["cycles"]
result["fx"] = g * np.sin(df["x"]/n*2*np.pi*f)
```
, bag_pack('gain', 100, 'cycles', 4)    //  dictionary of parameters
)
| render linechart 

Screenshot of sine demo showing query result.

Performance tips

  • Reduce the plugin's input dataset to the minimum amount required (columns/rows).
    • Use filters on the source dataset, when possible, with Kusto's query language.
    • To do a calculation on a subset of the source columns, project only those columns before invoking the plugin.
  • Use hint.distribution = per_node whenever the logic in your script is distributable.
  • Use Kusto's query language whenever possible, to implement the logic of your Python script.

Usage tips

  • To generate multi-line strings containing the Python script in your query editor, copy your Python script from your favorite Python editor (Jupyter, Visual Studio Code, PyCharm, and so on), paste it in your query editor, and then enclose the full script between lines containing three consecutive backticks. For example:

    ```
    python code
    ```

  • Use the externaldata operator to obtain the content of a script that you've stored in an external location, such as Azure Blob storage.

Example

    let script = 
        externaldata(script:string)
        [h'https://kustoscriptsamples.blob.core.windows.net/samples/python/sample_script.py']
        with(format = raw);
    range x from 1 to 360 step 1
    | evaluate python(
        typeof(*, fx:double),
        toscalar(script), 
        bag_pack('gain', 100, 'cycles', 4))
    | render linechart 

For more examples of UDF functions that use the Python plugin, see the Functions library.