如何在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()函数创建和操作二维数组。
