numpy数组的重塑和转置实现

import numpy as np

arr = np.array([1, 2, 3, 4, 5, 6, 7, 8])
# 将数组重塑为2行4列的形状
a = arr.reshape(2, 4)
# 将数组重塑为4行2列的形状
b = arr.reshape(4, 2)

print(a)
'''
[[1 2 3 4]
 [5 6 7 8]]
'''
print(b)
'''
[[1 2]
 [3 4]
 [5 6]
 [7 8]]
'''

import numpy as np

arr = np.array(
    [
        [1, 2, 3, 4],
        [5, 6, 7, 8],
        [9, 10, 11, 12]
    ]
)
# 将数组重塑为4行3列的形状
a = arr.reshape(4, 3)
# 将数组重塑为2行6列的形状
b = arr.reshape(2, 6)
print(a)
'''
[[ 1  2  3]
 [ 4  5  6]
 [ 7  8  9]
 [10 11 12]]
'''
print(b)
'''
[[ 1  2  3  4  5  6]
 [ 7  8  9 10 11 12]]
'''

import numpy as np

arr = np.array(
    [
        [1, 2, 3, 4],
        [5, 6, 7, 8],
        [9, 10, 11, 12]
    ]
)
# 将数组转换为一维数组
print(arr.flatten())
'''
[ 1  2  3  4  5  6  7  8  9 10 11 12]
'''
# 将数组转换为一维数组
print(arr.ravel())
'''
[ 1  2  3  4  5  6  7  8  9 10 11 12]
''' 

import numpy as np

arr = np.array(
    [
        [1, 2, 3, 4],
        [5, 6, 7, 8],
        [9, 10, 11, 12]
    ]
)
# 对数组进行转置
print(arr.T)
'''
[[ 1  5  9]
 [ 2  6 10]
 [ 3  7 11]
 [ 4  8 12]]
'''
# 对数组进行转置
print(arr.transpose())
'''
[[ 1  5  9]
 [ 2  6 10]
 [ 3  7 11]
 [ 4  8 12]]
'''