[Python] Clever way to write a nested for loop or N dimensional array as a string

I'm working on a tool to create inputs for array simulations, where various inputs are iterated over multiple series to create all combinations of inputs. Some values need to iterate coupled together, for a toy example, 'day':['Sun','Mon','Tue','Wed','Thu','Fri','Sat'] and 'class':[False,False,True,False,True,False,False] would be used in pairs, but 'week':[1,2,3,4,5] would be iterated in another dimension. I'm trying to fit all input lists and settings into a simple text format, to then be intrepeted by python and outputing all relevent combinations. If zip() is similar to an inner product in linear algebra, the function I'm trying to make is an outer product.

What I've tried is a dictionary of key:[iterable] pairs, and a list of "dimensions" to iterate through

inputs:
format_array = {
'year':['f2024','s2025'],
'week':[1,2,3,4,5],
'day':['Sun','Mon','Tue','Wed','Thu','Fri','Sat'],
'class':[False,False,True,False,True,False,False],
'course':['SciComp'],
'students':[['tmuzzy','jsmith'],['Alice','Bob','Charlie']]
}
format_dimensions = [1,2,3,3,0,1]


Where all keys that have the same dimension are iterated at the same time, (ie: year and list of students, and class and day) and have the same length, but each dimention is iterated over each combination. To do this I used the itertools.product function over a range object of the length of each dimension to get all index combinations: for dim in it.product(*[range(d) for d in dim_sizes]) Then the indexes get the values for each key:[iterable][index] combo to a dictionary and then save that to a list:

out_list = [
{'course': 'SciComp', 'year': 'f2024', 'week': 1, 'day': 'Sun', 'class': False, 'students': ['tmuzzy','jsmith']}
{'course': 'SciComp', 'year': 'f2024', 'week': 1, 'day': 'Mon', 'class': False, 'students': ['tmuzzy','jsmith']}
...
{'course': 'SciComp', 'year': 's2025', 'week': 5, 'day': 'Fri', 'class': False, 'students': ['Alice','Bob','Charlie']}
{'course': 'SciComp', 'year': 's2025', 'week': 5, 'day': 'Sat', 'class': False, 'students': ['Alice','Bob','Charlie']}
]


Which can then be easily called in a different function to create my array simulation:

for day in out_list:
print("In {course} during {year}, in week {week} on {day} is there class: {class} ".format(**day))


This works, but it does not allow multiple sizes of iterable, where the 'students' direction has a different length for every 'year', so it cannot be iterated independently and is passed out as a list This is a reasonably complicated problem, but a great solution would really improve my workflows. I would like suggestions to have a better way to setup the inputs to handle "uneven dimensions", but I can figure out the implementationdetails myself. I can provide the full current implementation if needed.

Continue reading...