将一维数组变为二维数组

将一维数组转换为二维数组可以使用numpy库中的reshape函数。reshape函数可以将一维数组转换为指定形状的二维数组。下面是一个例子：

import numpy as np

# 定义一维数组
arr1 = np.array([1, 2, 3, 4, 5, 6])

# 将一维数组转换为二维数组
arr2 = arr1.reshape(2, 3)

print(arr2)

输出结果为：

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

在上面的例子中，我们首先导入了numpy库，然后定义了一个一维数组arr1。接着，我们使用reshape函数将arr1转换为一个2行3列的二维数组arr2，并将结果打印出来。

相关问题

python将一维数组变成二维数组

将一维数组变成二维数组可以使用numpy库中的reshape()函数。假设原数组为a，我们可以使用以下代码将其变成一个5行2列的二维数组：

import numpy as np
a = np.array([1,2,3,4,5,6,7,8,9,10])
b = a.reshape(5,2)
print(b)

输出结果为：

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

其中，reshape()函数的参数表示新数组的形状，也就是新数组的行数和列数。在上面的例子中，a数组中有10个元素，我们将其变成了5行2列的二维数组。需要注意的是，新数组的元素个数必须和原数组的元素个数相同，否则会抛出ValueError异常。

一维数组变为二维数组

将一维数组变为二维数组需要知道原数组的长度和目标二维数组的行数和列数。可以通过循环遍历原数组，将元素按照规定的行列数放入新的二维数组中。

示例代码：

def reshape(nums, r, c):
    if r * c != len(nums):
        return nums
    res = [[0] * c for _ in range(r)]
    for i in range(len(nums)):
        res[i // c][i % c] = nums[i]
    return res