Create a 2D list out of 1D list

后端 未结 3 1398
梦如初夏
梦如初夏 2020-12-11 15:31

I am a bit new to Python and I want to convert a 1D list to a 2D list, given the width and length of this matrix.

Say I have a

3条回答
  •  被撕碎了的回忆
    2020-12-11 15:58

    I think you should use numpy, which is purpose-built for working with matrices/arrays, rather than a list of lists. That would look like this:

    >>> import numpy as np
    >>> list_ = [0,1,2,3]
    >>> a = np.array(list_).reshape(2,2)
    >>> a
    array([[0, 1],
           [2, 3]])
    >>> a.shape
    (2, 2)
    

    Avoid calling a variable list as it shadows the built-in name.

提交回复
热议问题