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

时间:2013-07-01 01:28:59

标签: python

我正在尝试在我的代码中执行类似的操作:

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)。我该如何解决这个问题?

2 个答案:

答案 0 :(得分:5)

试试这个:

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

fu()返回的unpack元组

答案 1 :(得分:0)

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