python - 如何从python中的结果中删除额外的列表

Question

结果有多余的括号，如何在 python 中删除它们？

我正在调用两个方法，其中一个返回字典元组，第二个返回字典元组列表。

print method_A() // ({'id': 6}, {'id': 9})
print method_B() // [({'id': 6}, {'id': 9})]

如何从第二种方法的结果中删除列表？

我尝试了类型检查并且成功了，但我想知道是否有任何简单的方法可以做到这一点。

我尝试了以下代码:

resultA = method_A() // ({'id': 6}, {'id': 9})
resultB method_B() // [({'id': 6}, {'id': 9})]

if type(resultA) == list:
    resultA = resultA[0]
if type(resultB) == list:
    resultB = resultB[0]

或者如果我知道，我可以直接使用 resultB[0]

Answer 1

如果这些方法之一总是返回一个包含元组元素的列表，只需使用索引:

resultA = method_A()
resultB = method_B()[0]

如果任一方法有时返回一个列表对象，有时只返回元组，请使用函数:

def unwrap(v):
    v[0] if isinstance(v, list) else v

并使用它:

resultA = unwrap(method_A())
resultB = unwrap(method_B())

然后联系创建这些方法的人，并与他们严肃地讨论 API 设计的一致性。