Python cells

Create a Python cell by either:

• Using the Y keyboard shortcut to place a new cell
• Selecting the Python cell option from the control bar
• Selecting the Add Python cell option when referencing a cell from the + icon that appears when the cell is selected.

How Python cells work

Python cells work very similarly to SQL cells, consisting of a text input area and an output area. Python cells are reactive just like all other cells, and their relationships are indicated by the same connector lines.

Python cells are executed on Count's servers in an isolated, sandboxed environment where network access can be controlled.

Referencing other cells using the cells variable

In Python cells, there is a special global cells variable that contains the results of other cells formatted as pandas DataFrames. Access cell results using keys or attributes on this object:

Most operations work the same way on categorical vs non-categorical columns, but if a non-categorical form is required, then use the astype method:

df['column'] = df['column'].astype('object')

Referencing Python variables

Each Python cell is run independently of any others. For this reason, variables defined at the root scope of one Python cell are not available in any other Python cell (they do not shared any scope).

Referencing Python cells from DuckDB

The last expression in a Python cell is special, and becomes the single output of that cell. If this output can be represented as a table, it can be queried by local DuckDB cells too:

Python cell outputs

As Python is a more expressive language than SQL, it is able to output more data types:

• Table output - if the final expression of the cell is representable as a table
• Image output - if the final expression of the cell is a PNG-formatted image bytes object
• Logs output - if the cell has printed anything during its execution

The output type defaults to Automatic, which can be overridden from the Output type button above the cell.

Interactivity

Because Python cells work just like any other cell, you can use control cells to add interactivity to any Python cell. In the example below, the parameters of a plot are adjustable using control cells:

HTML outputs

If a Python cell returns an object with an HTML representation, a View output button will appear. Clicking this button will cause the output to be displayed in the output section of the cell.

To close a HTML output, click the Close rich output button from the floating cell controls:

Technical details

An object is considered to have HTML output if:

• It has a method called _ipython_display_ which returns a string
• It has a method called _repr_html_ which returns a string
• It has a method called _repr_mimebundle_ which returns a dict with a key text/html

Any HTML output is contained within a sandboxed iframe, so some functionality may be restricted. Most existing packages which conform to the IPython standard methods described above should work with Count. If you encounter a package which does not work as expected, please contact Count support.

Python packages

Any public package on pypi.org should be available to use. To load a package, just import it as usual and Count will attempt to automatically download it and make it available. To use a specific version of a package you can either:

Use "Inline script metadata" (see here) to define dependency at the top of your cell like this:

# /// script
# dependencies = ["pandas==2.0.4"]
# ///

import pandas as pd

Or use Jupyter notebook style magic pip commands like this:

# %pip install pandas==2.0.4

import pandas as pd

Private Python packages are not currently supported.

Performing network requests

Network requests can be made by Python cells (e.g. using packages like requests and urllib3) provided it is allowed by your workspace. Specifically workspaces can be configured to:

• allow all network requests; or
• allow only network requests to specific domains (if no domains are specified then network requests are blocked completely)

Maximum request payload limits apply, so network requests will fail if they attempt to send too much data.

Using secrets

Network requests made by Python cells can also reference secrets (e.g. API tokens) as below:

import requests
from count import get_secret

response = requests.get(
    "https://example.com/path",
    headers={
        "Authorization": f"Bearer {get_secret("my_secret")}",
    },
)
data = response.json()

The Python code itself won't be able to access this secret. Instead get_secret("my_secret") returns a random token which will be substituted for the real secret as part of proxying the network request. This keeps your secret safe from being accidentally leaked.

Secrets can be configured at both the workspace and project levels (project secrets will take precedence if secrets share the same name). Secrets must be scoped to specific domains (even if network access is otherwise unrestricted).

There are some security considerations to note when using secrets in Python cells:

• Secrets can only be used in network requests.
• Secret values are inserted into the network requests and then sent to the URL specified in the request.
• Secret values are not accessible from the Count app regardless of your permission level.
• If you grant analyst access to a canvas, you should assume that the editor will be able to access any secret you have defined by, for example, sending it to a URL that they control.
• You should never directly enter a secret value into the text of a Python cell, as it will be visible to all viewers of that canvas, even if the cell input is hidden.

Be careful when using libraries which transform secrets. For example, instead of using the auth parameter in requests (which automatically base64-encodes its arguments), construct the Authorization header manually.