numpy中三维数组转变成二维数组

原文链接:https://www.cnblogs.com/zz22--/p/7647836.html

umpy中reshape()函数对三维数组进行转换成二维数组,见下面例子:

>>>a=np.reshape(np.arange(18),(3,3,2))
>>> a
array([[[ 0,  1],
        [ 2,  3],
        [ 4,  5]],

       [[ 6,  7],
        [ 8,  9],
        [10, 11]],

       [[12, 13],
        [14, 15],
        [16, 17]]])
>>>a=reshape(a,(-1,3))
>>>a
array([[ 0,  1,  2],
       [ 3,  4,  5],
       [ 6,  7,  8],
       [ 9, 10, 11],
       [12, 13, 14],
       [15, 16, 17]])

>>>a=np.reshape(np.arange(18),(2,3,3))
>>>a
array([[[ 0,  1,  2],
        [ 3,  4,  5],
        [ 6,  7,  8]],

       [[ 9, 10, 11],
        [12, 13, 14],
        [15, 16, 17]]])
>>>a=np.reshape(a,(-1,3))
>>>array([[ 0,  1,  2],
       [ 3,  4,  5],
       [ 6,  7,  8],
       [ 9, 10, 11],
       [12, 13, 14],
       [15, 16, 17]])

猜你喜欢

转载自blog.csdn.net/ch206265/article/details/106976606
今日推荐