我正在尝试查找包含其他列表或元组的列表或元组的最大值。我首先想到的是将其弄平,然后使用max()函数从整个列表中找到最大值,但是我很难做到这一点。 有什么建议吗?
例如,您有一个元组:(1、2、3,(1、2)),预期输出为3
另一个示例是列表[1,(2、3),[4、5]],输出应为5
这需要两个使用展平函数并调用它:
def flatten(t):
output = []
for item in t:
if type(item) != type([]) or type(()):
output.append(item)
else:
output.extend(flatten(item))
return output
def max_val(t):
flatten(t)
return max(output)
答案 0 :(得分:1)
您可以使用chain.from_iterable
和collections.Iterable
定义通用函数来拼合输入:
from itertools import chain
from collections import Iterable
x = (1, 2, 3, (1, 2))
y = [1, (2, 3), [4, 5]]
def flatten(x):
return chain.from_iterable([i] if not isinstance(i, Iterable) else i for i in x)
res = max(flatten(x)) # 3
res = max(flatten(y)) # 5
答案 1 :(得分:0)
您可以使用more_itertools包来展平您所拥有的东西。
import more_itertools
lst = [[1, 2, 3], [4, 5, 6], [7], [8, 9]]
res = list(more_itertools.flatten(lst))
然后,您只需要使用max()函数。
答案 2 :(得分:0)
您可以创建自定义函数以使其可迭代性变平坦,并使用标准的max
或min
函数:
from collections.abc import Iterable
x = (1, 2, 3, (1, 2))
def my_flatten(iterable):
for value in iterable:
if isinstance(value, Iterable):
yield from my_flatten(value)
else:
yield value
print('min = ', min(my_flatten(x)))
print('max = ', max(my_flatten(x)))
输出:
min = 1
max = 3