from typing import Dict


def scale_z_axis(neurons: Dict, z_scale: float) -> Dict:
    """
    Ensure z axis is the same unit with xoy plate.
    :param neurons: [xmin, ymin, xmax, ymax, z] style
    :param z_scale: scale factor
    :return: [xmin, ymin, xmax, ymax, z * scale] style
    """

    for key, value in neurons.items():
        neurons[key] = [[bb[0], bb[1], bb[2], bb[3], bb[4] * z_scale] for bb in value]  # Only a calculation on z value to reduce time consumption.
    return neurons