📅  最后修改于: 2023-12-03 14:51:05.735000             🧑  作者: Mango
在 Python 中初始化二维数组
Python 中的列表可以看作是动态数组,列表可以包含任何 Python 对象,包括其他列表(也称为嵌套列表)。可以使用列表创建和初始化二维数组。本文将介绍几种方式来初始化二维数组,代码片段需按 markdown 标明。
利用嵌套列表创建二维数组是 Python 中最常见的方法,使用双重循环来初始化二维数组。
# 创建5行3列的二维数组
rows = 5
cols = 3
matrix = []
for i in range(rows):
row = []
for j in range(cols):
row.append(0)
matrix.append(row)
print(matrix)
输出:
[[0, 0, 0], [0, 0, 0], [0, 0, 0], [0, 0, 0], [0, 0, 0]]
利用列表推导式创建二维数组是 Python 中比较简便的方法,代码量较少,但是可读性稍差。
# 创建5行3列的二维数组
rows = 5
cols = 3
matrix = [[0 for j in range(cols)] for i in range(rows)]
print(matrix)
输出:
[[0, 0, 0], [0, 0, 0], [0, 0, 0], [0, 0, 0], [0, 0, 0]]
利用 NumPy 库创建二维数组是 Python 中较为高级的方法,NumPy 库不仅仅支持二维数组,也支持任意维度的数组,该库提供了高效的数学库和科学计算库,可用于一些数据科学领域的操作。
# 导入 NumPy 库
import numpy as np
# 创建5行3列的二维数组
rows = 5
cols = 3
matrix = np.zeros((rows, cols))
print(matrix)
输出:
[[0. 0. 0.]
[0. 0. 0.]
[0. 0. 0.]
[0. 0. 0.]
[0. 0. 0.]]
总结:
对于小规模的二维数组,建议使用第一种或第二种方法初始化,对于大规模的数组或需要进行科学计算的数组数据,建议使用第三种方法初始化。