zarr-python 3 array creation example

create_sharded_array.py

# /// script
# requires-python = ">=3.11"
# dependencies = [
#     "zarr == 3.0.0",
# ]
# ///

from zarr import create_array
import numpy as np

data = np.arange(1024 ** 3).reshape((1024,) * 3).astype('uint8')
arr = create_array(
    'path/to/data.zarr', # or s3://bucket/path/to/data.zarr
    shape=data.shape,
    dtype=data.dtype,
    shards=data.shape, # just 1 shard
    chunks=(64,) * 3, # nice small chunks
    compressors='auto', # use default compression
    )

arr[:] = data # fill the array with valus
print(arr.info_complete())