欢迎访问宙启技术站
智能推送

如何在Python中使用array()函数创建二维数组

发布时间:2023-12-26 17:35:25

在Python中,我们可以使用NumPy库中的array()函数来创建二维数组。array()函数接受一个列表作为参数,其中每个元素都是一个包含相同长度的子列表。

下面是一个使用array()函数创建二维数组的例子:

import numpy as np

# 创建一个包含3行4列的二维数组
arr = np.array([[1, 2, 3, 4],
                [5, 6, 7, 8],
                [9, 10, 11, 12]])

print(arr)

输出:

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

在这个例子中,我们创建了一个包含3行4列的二维数组。每个子列表代表数组的一行。输出结果显示了二维数组的内容。

我们也可以通过数组的“shape”属性获取二维数组的形状:

print(arr.shape)

输出:

(3, 4)

这说明数组有3行和4列。

我们还可以使用索引来访问和修改二维数组的元素。索引从0开始, 个索引表示行数,第二个索引表示列数。例如,要访问第二行第三列的元素,我们可以使用以下代码:

print(arr[1, 2])

输出:

7

要修改这个元素的值,我们可以使用赋值运算符:

arr[1, 2] = 10
print(arr)

输出:

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

在这个例子中,我们将第二行第三列的元素更新为10。

除了创建固定大小的二维数组之外,我们还可以使用numpy库的其他函数创建特殊类型的二维数组,比如全零数组、全1数组和单位矩阵。例如,以下代码创建了一个3行4列的全零数组:

arr = np.zeros((3, 4))
print(arr)

输出:

[[0. 0. 0. 0.]
 [0. 0. 0. 0.]
 [0. 0. 0. 0.]]

以上是使用array()函数在Python中创建二维数组的简单示例。希望可以帮助你理解如何使用array()函数创建和操作二维数组。