python - 在另一个函数调用中使用返回元组的 Python 函数

Question

我试图在我的代码中做这样的事情：

def fu():
    return np.array([1,2,3]), np.array([4,5,6])

def bar(x,y,z):
    print np.size(x)
    print np.size(y)
    print np.size(z)

bar(np.array([7,8]), fu())

但我收到一条错误消息说bar() takes exactly 3 arguments (2 given)。我怎么解决这个问题？

Answer 1

试试这个：

bar(np.array([7,8]), *fu())

（解压返回的元组fu()）

Answer 2

import numpy as np

def fu():
   return np.array([1,2,3]), np.array([4,5,6])

def bar(x,y,z):
   print np.size(x)
   print np.size(y)
   print np.size(z)

bar(np.array([7,8]), *fu())

--output:--
2
3
3