从NumPy函数返回一个数组而不是一个元组

Question

我有以下代码：

return atleast_1d(0.4*P2), atleast_1d(k_uni)

我的函数返回一个元组。以数组形式返回的最快方法是什么？

Answer 1

此return语句创建一个元组，由2个数组组成。

return atleast_1d(0.4*P2), atleast_1d(k_uni)

如果你做了

，你会得到一个元组

return 1, 2
return [1,2,3],{'one':1}
etc

我建议用

调用该函数

x, y = foo(...)

现在，return语句中定义的2个数组被分配给2个变量。你可以用任何有效的方式组合它们（取决于它们的形状）。

np.array((x,y))
np.concatenate((x,y))
etc

请注意，我实际上是在为这些函数提供一个元组。我本可以把它写成一个清单。

问题不在于函数如何返回元组，而在于如何将2个数组合并为一个。