博客

使用NumPy在Python中扁平化一个矩阵的步骤如下：

1. 导入NumPy模块

import numpy as np

1. 创建一个矩阵

matrix = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

1. 使用flatten()方法扁平化矩阵

flattened_matrix = matrix.flatten()

1. 输出结果

print(flattened_matrix)

示例1：以一个4×4的矩阵为例

import numpy as np

matrix = np.array([[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12], [13, 14, 15, 16]])
flattened_matrix = matrix.flatten()

print(flattened_matrix)

输出结果：

[ 1  2  3  4  5  6  7  8  9 10 11 12 13 14 15 16]

示例2：以一个3x2x2的矩阵为例

import numpy as np

matrix = np.array([[[1, 2], [3, 4]], [[5, 6], [7, 8]], [[9, 10], [11, 12]]])
flattened_matrix = matrix.flatten()

print(flattened_matrix)

输出结果：

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

使用NumPy中的flatten()方法可以轻松将任何形状的矩阵扁平化为一维序列。