convert.py 文件源码

def _master_dataarray(exp, data_dict):
    case_list = [exp._case_data[case] for case in exp.cases]
    stacked_data = _stack_dims(data_dict, case_list, {}, exp)

    test_case = next(exp.all_cases())
    test_da = data_dict[test_case]
    name = test_da.name

    new_coords = test_da.to_dataset().coords
    logger.debug("Creating master dataarray")
    for case in exp.cases:
        logger.debug("   " + case)
        new_coords[case] = exp._case_data[case].vals

    new_dims = list(exp.cases) + list(test_da.dims)

    new_da = DataArray(stacked_data, coords=new_coords,
                       dims=new_dims)
    new_da = copy_attrs(test_da, new_da)
    new_da.name = name

    return new_da