kvp_to_array.py

# Convert a list of (k,v) pairs into an array
def kvp_to_array(dims, data, ind=0, baseline=0):
    vol = np.zeros(dims,dtype=data[0][1].dtype) + np.array(baseline).astype(data[0][1].dtype)
    
    for k,v in data:
        # check if data contains a single value or an iterable
        if hasattr(v,'__iter__'):
            vol[k] = v[ind]
        else:
            vol[k] = v
    return vol