如何将一维图元数组转换为二维Numpy数组

  • Post category:Python

将一维图元数组转换为二维Numpy数组可以通过reshape函数实现。具体步骤如下:

1.导入numpy模块

import numpy as np

2.创建一维图元数组

下面以一个包含9个元素的一维图元数组为例:

image_array = [1,2,3,4,5,6,7,8,9]

3.使用reshape函数将一维图元数组转换为二维Numpy数组

使用reshape函数,将一维数组转换为二维数组。函数的参数为元组,元组中的第一个值为行数,第二个值为列数。如下所示:

image = np.array(image_array).reshape(3,3)

4.输出结果

可以通过print函数来查看转换后的二维Numpy数组,如下所示:

print(image)

输出结果如下:

array([[1, 2, 3],
       [4, 5, 6],
       [7, 8, 9]])

示例1:

假设有一个一维图元数组包含16个元素,我们想将其转换为4行4列的二维Numpy数组,代码如下:

image_array = [1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16]
image = np.array(image_array).reshape(4,4)
print(image)

输出结果如下:

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

示例2:

假设有一个一维图元数组包含6个元素,我们想将其转换为2行3列的二维Numpy数组,代码如下:

image_array = [1,2,3,4,5,6]
image = np.array(image_array).reshape(2,3)
print(image)

输出结果如下:

array([[1, 2, 3],
       [4, 5, 6]])

以上就是将一维图元数组转换为二维Numpy数组的完整攻略,需要注意reshape函数参数为元组,元组中的第一个值为行数,第二个值为列数。