如何根据参数n动态生成reshape()函数的重复参数对？

这个需求其实很好实现，核心思路就是动态构造reshape需要的参数元组——因为你要的参数规律非常明确：n值是多少，就把(nrows, ncols)这个子组重复多少次。

具体实现代码 #

在Python（尤其是用NumPy的场景）里，元组支持直接乘以整数来实现重复操作，我们可以写一个简单的函数封装这个逻辑：

import numpy as np

def dynamic_reshape(results, n, nrows, ncols):
    # 生成重复n次的reshape参数元组
    reshape_dims = (nrows, ncols) * n
    # 调用reshape，直接传入构造好的元组即可
    return results.reshape(reshape_dims)

调用示例 #

比如当你需要n=2时：

# 假设results是符合长度要求的NumPy数组
arr_multi_dim = dynamic_reshape(results, n=2, nrows=5, ncols=5)
# 等价于手动编写：results.reshape(5,5,5,5)

当n=4时：

arr_multi_dim = dynamic_reshape(results, n=4, nrows=5, ncols=5)
# 等价于手动编写：results.reshape(5,5,5,5,5,5,5,5)

重要注意事项 #

别忘了NumPy的reshape有个硬性要求：数组的总元素数必须等于所有维度的乘积，也就是：
results.size == (nrows * ncols) ** n
如果不满足这个条件，会直接抛出ValueError，所以使用前要确保你的results数组长度是符合要求的哦。

内容的提问来源于stack exchange，提问作者Leockl