create_named_matrix

create_named_matrix.py

import more_itertools
import itertools
"""
>>> create_named_matrix(2,3,5)
array([[['0,0,0', '0,0,1', '0,0,2', '0,0,3', '0,0,4'],
        ['0,1,0', '0,1,1', '0,1,2', '0,1,3', '0,1,4'],
        ['0,2,0', '0,2,1', '0,2,2', '0,2,3', '0,2,4']],

       [['1,0,0', '1,0,1', '1,0,2', '1,0,3', '1,0,4'],
        ['1,1,0', '1,1,1', '1,1,2', '1,1,3', '1,1,4'],
        ['1,2,0', '1,2,1', '1,2,2', '1,2,3', '1,2,4']]], dtype=object)
"""
def create_named_matrix(*dims):
    dim_ranges = [range(i) for i in dims]
    matrix = list(itertools.product(*dim_ranges))
    matrix = list(map(lambda x: ",".join(map(str,x)),matrix))
    matrix = np.array(matrix,dtype=object).reshape(*dims)
    return matrix