如何在Python中把一维数组作为列转换成二维数组

  • Post category:Python

将一维数组转为二维数组实际上就是将一维数组中的每个元素作为二维数组中的一列,需要用到 numpy 库提供的reshape()函数,下面给出具体的步骤和示例。

步骤如下:

  1. 导入 numpy 库。

  2. 定义一维数组。

  3. 使用 reshape() 函数将一维数组转换为二维数组。

  4. 查看转换后的二维数组。

下面是两个示例,第一个示例是将列表转换为二维数组,第二个示例是将字符串(每个字符作为一个元素)转换为二维数组。

示例1:将列表转换为二维数组

import numpy as np

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

arr2 = np.array(arr1).reshape(3, 3)

print(arr2)

输出结果:

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

示例2:将字符串转换为二维数组

import numpy as np

str1 = 'Hello World'

arr1 = np.array(list(str1))

arr2 = arr1.reshape(2, 6)

print(arr2)

输出结果:

array([['H', 'e', 'l', 'l', 'o', ' '],
       ['W', 'o', 'r', 'l', 'd']],
      dtype='<U1')

在以上示例中,我们先定义了一个一维数组(或字符串),然后使用 np.array() 将其转换为数组,再使用 reshape() 函数将其转换为二维数组,最后打印输出。注意需要指定二维数组的行列数,确保一维数组中的元素能够平均分布到二维数组的每一列上。

希望这个攻略对你有所帮助!