tessl/pypi-zarr

An implementation of chunked, compressed, N-dimensional arrays for Python

Overview

Eval results

Files

Data I/O Operations

Name: tessl/pypi-zarr
Author: tessl

High-level functions for saving and loading zarr data structures to and from storage. These provide convenient interfaces for persistence operations with support for various data formats and storage backends.

Capabilities

Data Loading

def load(
    store: StoreLike,
    path: str = None,
    **kwargs
) -> Any

Load zarr array data into memory as a numpy array or nested structure.

Parameters:

store: Storage location containing zarr data
path: Path within store to load from

Returns: Numpy array for single arrays, dict-like structure for groups

Single Array Saving

def save_array(
    store: StoreLike,
    arr: ArrayLike, 
    path: str = None,
    **kwargs
) -> None

Save a single array to zarr format.

Parameters:

store: Storage location to save to
arr: Array data to save
path: Path within store to save to

Group Saving

def save_group(
    store: StoreLike,
    *args,
    path: str = None,
    **kwargs
) -> None

Save a group structure with multiple arrays.

General Save Function

def save(
    file: Union[str, StoreLike],
    *args,
    **kwargs
) -> None

General-purpose save function that handles both arrays and groups.

Parameters:

file: File path or storage location
*args: Arrays or array-like data to save
**kwargs: Additional arrays as keyword arguments

Usage Examples

Basic Array I/O

import zarr
import numpy as np

# Create and save array
data = np.random.random((1000, 1000))
zarr.save('dataset.zarr', data)

# Load array back
loaded_data = zarr.load('dataset.zarr')

Multiple Array I/O

# Save multiple arrays
arr1 = np.random.random((500, 500))
arr2 = np.random.random((300, 300))

zarr.save('multi_arrays.zarr', 
          temperature=arr1, 
          humidity=arr2)

# Load returns dict-like structure
data = zarr.load('multi_arrays.zarr')
temp = data['temperature']
humidity = data['humidity']

Advanced I/O with Compression

from zarr.codecs import BloscCodec

# Save with compression
zarr.save('compressed.zarr', 
          data, 
          compressor=BloscCodec(cname='zstd', clevel=3))

# Load automatically handles decompression
loaded = zarr.load('compressed.zarr')

Install with Tessl CLI