python list comprehension for multiple return function to numpy array

Question

从这个线程我们可以看到，如何将多个返回值从列表理解解包到元组。

def f(x):
  return 2*x, x*x

x = range(1, 4)
xlist, ylist = zip(*[f(value) for value in x])

有没有办法直接对 numpy 数组执行此操作？或者是解决此问题的唯一方法:

xlist = np.asarray(xlist)

?

Answer 1

要使用 numpy.array 执行此操作，只需将参数作为数组本身传递即可。函数中的操作被映射到数组参数上，并返回两个数组的元组:

>>> xlist, ylist = f(np.asarray(x))
>>> xlist
array([2, 4, 6])
>>> ylist
array([1, 4, 9])