展平列表或元组,然后在Python中找到最大值?

时间:2018-07-15 11:56:27

标签: python

我正在尝试查找包含其他列表或元组的列表或元组的最大值。我首先想到的是将其弄平,然后使用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)

3 个答案:

答案 0 :(得分:1)

您可以使用chain.from_iterablecollections.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)

您可以创建自定义函数以使其可迭代性变平坦,并使用标准的maxmin函数:

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