def total_length(nrn_pop, neurite_type=NeuriteType.all):
"""Get the total length of all sections in the group of neurons or neurites."""
nrns = neuronfunc.neuron_population(nrn_pop)
return list(sum(section_lengths(n, neurite_type=neurite_type)) for n in nrns)
def map_neurons(fun, neurites, neurite_type):
"""Map `fun` to all the neurites in a single or collection of neurons."""
nrns = neuronfunc.neuron_population(neurites)
return [fun(n, neurite_type=neurite_type) for n in nrns]