How to remove the innermost level of nesting in a list of lists of varying lengths

前端 未结 7 1480
感情败类
感情败类 2021-01-13 14:51

I\'m trying to remove the innermost nesting in a list of lists of single element length lists. Do you know a relatively easy way (converting to NumPy arrays is fine) to get

7条回答
  •  攒了一身酷
    2021-01-13 15:04

    How about np.squeeze?

    Remove single-dimensional entries from the shape of an array.

    arr = [ [ [1],[2],[3],[4], [5] ], [ [6],[7],[8] ] , [ [11],[12] ] ]
    >>> arr
    [[[1], [2], [3], [4], [5]], [[6], [7], [8]], [[11], [12]]]
    >>> [np.squeeze(i) for i in arr]
    [array([1, 2, 3, 4, 5]), array([6, 7, 8]), array([11, 12])]
    

    Not necessarily the innermost (ie independent of how many dimensions) dimension though. But your question specifies "list of lists"

提交回复
热议问题