我需要在转置嵌套列表中找到最小值和最大值,而忽略任何none类型。
这是我的嵌套列表:
x = [[1, 20, 50],
[5, 6, 7],
[11, 42, 2],
[7, 32, None]]
我想忽略第三栏中的“无”,并希望获得以下输出:
min
[1, 6, 2]
max
[11,42,50]
我需要使用标准的python库
答案 0 :(得分:2)
纯python解决方案:
In [16]: x = [[1, 20, 50],
...: [5, 6, 7],
...: [11, 42, 2],
...: [7, 32, None]]
...:
In [17]: [min((y for y in x if y is not None), default=None) for x in zip(*x)]
Out[17]: [1, 6, 2]
In [18]: [max((y for y in x if y is not None), default=None) for x in zip(*x)]
Out[18]: [11, 42, 50]
请注意,对于[[None]]
,上面的代码将返回[None]
,因为既没有最小元素也没有最大元素。如果您希望此代码引发异常,只需删除default=None
。如果您想从结果列表中排除None
,只需使用[z for z in (...) if z is not None]
Numpy解决方案,将其强制转换为float以将None自动转换为nan:
In [12]: import numpy as np
In [13]: a = np.array(
...: [[1, 20, 50],
...: [5, 6, 7],
...: [11, 42, 2],
...: [7, 32, None]],
...: dtype=np.float)
...:
In [14]: np.nanmin(a, axis=0).astype(np.int)
Out[14]: array([1, 6, 2])
In [15]: np.nanmax(a, axis=0).astype(np.int)
Out[15]: array([11, 42, 50])