使用array函数可以创建二维数组,只需将一个嵌套列表作为参数传给该函数即可。下面的代码创建一个二维数组,该数组的第一维包含两个元素(即两个子列表),第二维包含3个元素(即每个列表中的3个数字)。如果将该数组看作表格,则该表格有2行3列。
np.array([[1,2,3],[7,8,9]])
上述代码将创建以下数组:
[[1 2 3] [7 8 9]]
使用NumPy中的random函数可以创建一个包含0~1的随机数的二维数组,该函数位于NumPy库的random模块中。将一个元组作为参数传递给random函数,该元组中的值表示二维数组的行数和列数。下面的代码是创建一个2行3列的二维数组。
np.random.random((2,3))
创建的数组如下,每次运行相同的代码会得到不同的随机数。
[[0.53631257 0.02745073 0.15569723] [0.72433551 0.36637602 0.13712715]]