python - 从python函数分配可变数量的返回的优雅方法

Question

我正在使用一个带有函数的 python 库，它以列表的形式返回结果。根据实验配置，此列表由一个或两个元素组成。在这种情况下，分配函数输出的简单方法如下:

bbox, segm = None, None
results = test_model()  # returns a list with either one or two elements
bbox = results[0]
if len(results) > 1:
    segm = results[1]

# [some other code here]
if segm is not None:
    plot(segm)

但是，这似乎很冗长，因为我们需要先将 bbox 和 segm 都初始化为 None，然后评估 if len(results) > 1.有什么pythonic方法可以避免这种情况吗？理想情况下，这样的事情会非常好:

bbox, segm = test_model()  # returns a list with either one or two elements
if segm is not None:
    plot(segm)

Answer 1

您可以使用 structural pattern matching ，在 Python 3.10 中可用:

match test_model():
   case [bbox, segm]:
      plot(segm)
   case bbox:
      pass

但是，如果 test_module 总是返回一个元组，你可以使用解包:

bbox, *segm = test_model()

现在，如果 text_model 返回一个值，segm 将是一个空列表 ([])。但是，如果它返回两个值，segm 将包含单个元素。因此，您的完整代码可以变成:

bbox, *segm = test_model()
if segm:
   plot(segm[0])