将功能1的输出用作功能2的输入

Question

所以我有一些类似以下的代码：

def _step_1(a, b, c):
    some codes
    return d, e, f

def _step_2(d, e, f, a, b):
    some codes
    return g

def _run_all(a, b, c):
    g = _step_2(_step_1(a, b, c), a, b)
    return g

它告诉我，我缺少两个参数“ a”和“ b”。有人可以告诉我是否通过尝试节省一些步骤做错了什么？还是没有办法节省步骤？我知道我绝对可以这样写：

def _run_all(a, b, c):
    d, e, f = _step_1(a, b, c)
    g = _step_2(d, e, f, a, b)
    return g

Answer 1

如果您的版本是python 3，请使用解压缩（*）：

def _run_all(a, b, c):
  g = _step_2(*_step_1(a, b, c), a, b)
  return g