Python 询问要在Numpy中运行代码的发行版

我有一个依赖于概率分布的代码。默认情况下，我使用从正态分布采样的Numpy函数np.random.normal。所以这个Numpy包的结构是np随机分布

如何编辑代码以询问用户希望在哪个发行版上运行的输入。它应该是这样工作的：

应向用户提供np.random包中所有可用分布的列表

用户将选择一个（选择将被枚举，因此用户将为某些分发选择例如数字5）

用户会被询问他们希望为所选的分发设置哪些参数

第3行。以上是困扰我的问题。包np.random包含大约40个分布（按链接），技术上我可以为每个分布手动键入公式，该公式将根据用户选择的内容运行。然而，大多数分布都有特定的参数，例如，正态分布需要3个输入，而几何分布需要2个输入。因此，如果用户选择正态分布，那么他们需要指定这3个参数，对于几何图形也是如此，但在这种情况下，需要指定2个参数

有没有更好的更“自动”的方法