numpy系列之数组的重塑

474 阅读2分钟

1.数组重塑

所谓数组重塑就是更改数组的形状。比如将原来3行4列的数组重塑成4行3列的数组。在numpy中用reshape方法来实现数组重塑

1.1 一维数组重塑

一维数组重塑就是将数组从一行或一列数组重塑为多行多列的数组。

先创建一个一维数组

import numpy as np
​
arr = np.arange(8)
print(arr)

result:

[0 1 2 3 4 5 6 7]

上面的这个数组既可以转换为2行4列的多维数组, 也可以转换为4行2列的多维数组

1.1.1 将数组重塑为2行4列的多维数组

print(arr.reshape(2, 4))

result:

[[0 1 2 3]
 [4 5 6 7]]

1.1.2 将数组重塑为4行2列的多维数组

print(arr.reshape(4, 2))

result:

[[0 1]
 [2 3]
 [4 5]
 [6 7]]

注:无论2行4列还是4行2列,只要重塑后数组中的值的个数等于重塑前一维数组中的值的个数即可。

1.2 多维数组重塑

先创建一个多维数组

import numpy as np
​
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9], [10, 11, 12]])
print(arr)

result:

[[ 1  2  3]
 [ 4  5  6]
 [ 7  8  9]
 [10 11 12]]

同样地,上面的这个数组既可以转换为3行4列的多维数组, 也可以转换为2行6列的多维数组

1.2.1 将数组重塑为3行4列的多维数组

print(arr.reshape(3, 4))

result:

[[ 1  2  3  4]
 [ 5  6  7  8]
 [ 9 10 11 12]]

1.2.2 将数组重塑为2行6列的多维数组

print(arr.reshape(2, 6))

result:

[[ 1  2  3  4  5  6]
 [ 7  8  9 10 11 12]]

注:我们同样可以将4行3列的多维数组重塑为3行4列或者2行6列的多维数组,只要重塑后数组中的值的个数等于重塑前一维数组中的值的个数即可。

2.数组转置

数组转置就是将数组的行旋转为列,用到的方法是.T。这里可以将转置看做是一种特殊的重塑。

import numpy as np
​
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9], [10, 11, 12]])
print(arr)

result:

[[ 1  2  3]
 [ 4  5  6]
 [ 7  8  9]
 [10 11 12]]
print(arr.T)

result:

[[ 1  4  7 10]
 [ 2  5  8 11]
 [ 3  6  9 12]]