合并嵌套列表并查找ord

Question

我有这样一个嵌套列表：

[
 [
  [0.08, 0.34,0.68],
  [0.20, 0.5,  0.92, 0.96],
  [0.23, 0.52, 0.88, 0.91],
  [inf]
 ],
 [
  [0.60],
  [inf],
  [inf],
  [0.05, 0.70, 0.79, 0.89]
 ]
]

例如，一级向下的大小是固定的7x2。但最底层是可变的。你知道吗

我想创建一个这样的排序组合列表

[0.05,0.08,0.20,0.23,0.34,...]

但是我还想要数字所属的列表，例如0.05属于[1,3]等等。你知道吗

有人能帮我吗？我可以把它组合成一个排序的列表。但是我找不到有效的方法来获取列表索引。你知道吗

Answer 1

像这样的？你知道吗

from numpy import inf
lst = [[[0.08, 0.34,0.68],[0.20, 0.5, 0.92, 0.96], [0.23, 0.52, 0.88, 0.91], [inf]], [[0.60], [inf], [inf],[0.05, 0.70, 0.79, 0.89]]]

sorted((e, i, j) for i in range(len(lst)) for j in range(len(lst[0])) for e in lst[i][j])

# [(0.05, 1, 3),
#  (0.08, 0, 0),
#  (0.2, 0, 1),
#  (0.23, 0, 2),
#  (0.34, 0, 0),
#  (0.5, 0, 1),
# ...

您可以使用zip从值中解压索引：

vals, *inds = zip(*sorted((e, i, j) for i in range(len(lst)) for j in range(len(lst[0])) for e in lst[i][j]))

vals
# (0.05, 0.08, 0.2, 0.23, 0.34, 0.5, 0.52, 0.6, 0.68, 0.7, 0.79, 0.88, 0.89, 0.91, 0.92, 0.96, inf, inf, inf)

inds
# [(1, 0, 0, 0, 0, 0, 0, 1, 0, 1, 1, 0, 1, 0, 0, 0, 0, 1, 1),
#  (3, 0, 1, 2, 0, 1, 2, 0, 0, 3, 3, 2, 3, 2, 1, 1, 3, 1, 2)]

Answer 2

您可以递归地分析嵌套到任何深度的列表，方法如下：

def nested_list_walker(nested_lists, prefix=()):
    """Yields item, nested_index pairs."""
    try:
        for index, nested_list in enumerate(nested_lists):
            for output in nested_list_walker(nested_list,
                                             prefix=prefix + (index,)):
                yield output
    except TypeError:
        yield nested_lists, prefix

在您的示例中，对该生成器进行排序会产生：

[(0.08, (0, 0, 0)),
 (0.2, (0, 1, 0)),
 (0.23, (0, 2, 0)),
 (0.34, (0, 0, 1)),
 (0.5, (0, 1, 1)),
 (0.52, (0, 2, 1)),
 (0.6, (1, 0, 0)),
 (0.68, (0, 0, 2)),
 (0.7, (1, 3, 1)),
 (0.79, (1, 3, 2)),
 (0.88, (0, 2, 2)),
 (0.89, (1, 3, 3)),
 (0.91, (0, 2, 3)),
 (0.92, (0, 1, 2)),
 (0.96, (0, 1, 3)),
 (inf, (0, 3, 0)),
 (inf, (1, 1, 0)),
 (inf, (1, 2, 0))]