首页 - 安全 - python 矩阵转换数组

python 矩阵转换数组

2023-10-10 23:59

在python中,矩阵是一种常用的数据结构。在某些场景下,需要将矩阵转换为数组,此时就需要使用python的numpy库。

将 numpy 导入为 np
matrix = np.array([[1,2,3],[4,5,6],[7,8,9]]) #定义一个3x3矩阵
array = matrix.flatten() #将矩阵展开为一维数组
print(array)

上面的代码中,首先导入了numpy库,然后定义了一个3x3的矩阵。 numpy库的flatten()方法用于将矩阵展开为一维数组并打印出来。

如果要将多维数组展平为一维数组,可以使用numpy库的ravel()方法。代码如下:

将 numpy 导入为 np
matrix = np.array([[[1,2],[3,4]],[[5,6],[7,8]]]) #定义一个2x2x2多维数组
array = matrix.ravel() #将多维数组展开为一维数组
print(array)

上面的代码中,首先导入了numpy库,然后定义了一个2x2x2的多维数组。使用numpy库的ravel()方法将多维数组扩展为一维数组并打印。出来。