插入numpy数组的动态/参数化维度

I want to write a function which inserts matrices along a dimension into an numpy array. Where the dimension is given as a parameter to the function. However, I am struggling with the fact that the dimension shall be dynamic. If I'd know the dimension something like this would work:I want to write a function which inserts matric