python - 如何为输入列表重复一个函数？

Question

我正在尝试使用 QuTip 库的以下功能:

coherent_dm(N=10, alpha = c, offset=0, method='operator')

它的输出是一个矩阵，它的输入是一个复数“c”，但是我想获取多个“c”数字列表的矩阵，但是如果我将“c”定义为:

t = np.linspace(0,2*np.pi,100)
c = 2*np.exp(-t*1j) + 0.1*10*(1 - np.exp(-t*1j))

并运行它给我“尺寸不匹配”错误的代码。在这种情况下，我尝试运行一个包含 100 个复数的列表作为输入“c”，期望输出 100 个矩阵。任何有关如何解决此问题的想法将不胜感激。

Answer 1

coherent_dm 返回 qutip.Qobj 实例。这是一个满足 __array__ 的特殊对象。接口(interface)，但实际上并不是一个 numpy 数组。在内部，它似乎包含一个稀疏矩阵。这意味着你不能轻易地制作 coherent_dm一次返回多个矩阵，甚至将多个调用的结果连接到一个数组中。
因此，您最好的选择可能是使用列表推导:

result = [coherent_dm(N=10, alpha=i, offset=0, method='operator') for i in c]