创建二维数组的办法

直接创建(不推荐)

使用模块numpy创建

举个栗子:

创建一个3*3矩阵,并计算主对角线元素之和。
import numpy as np
a=np.random.randint(1,100,9).reshape(3,3) #生成形状为9的一维整数数组
a=np.random.randint(1,100,(3,3)) #上面的语句也可以改为这个
print(a)
(m,n)=np.shape(a) # (m,n)=(3,3)
sum=0
for i in range(m):
for j in range(n):
if i==j:
sum+=a[i,j]
print(sum)
其中一次输出:
[[96 42 16]
[19 14 92]
[39 29 95]]205
numpy中random:
numpy.random.randint(low, high=None, size=None, dtype='l'):生成一个整数或N维整数数组,取数范围:若high不为None时,取[low,high)之间随机整数,否则取值[0,low)之间随机整数。
#numpy.random.randint(low, high=None, size=None, dtype='l')
importnumpy as np#low=2
np.random.randint(2)#生成一个[0,2)之间随机整数#low=2,size=5
np.random.randint(2,size=5)#array([0, 1, 1, 0, 1])#low=2,high=2#np.random.randint(2,2)#报错,high必须大于low#low=2,high=6
np.random.randint(2,6)#生成一个[2,6)之间随机整数#low=2,high=6,size=5
np.random.randint(2,6,size=5)#生成形状为5的一维整数数组#size为整数元组
np.random.randint(2,size=(2,3))#生成一个2x3整数数组,取数范围:[0,2)随机整数
np.random.randint(2,6,(2,3))#生成一个2x3整数数组,取值范围:[2,6)随机整数#dtype参数:只能是int类型
np.random.randint(2,dtype='int32')
np.random.randint(2,dtype=np.int32)
reshape()用法
arange()用于生成一维数组
reshape()将一维数组转换为多维数组

举个栗子:

importnumpy as npprint('默认一维为数组:', np.arange(5))print('自定义起点一维数组:',np.arange(1, 5))print('自定义起点步长一维数组:',np.arange(2, 10, 2))print('二维数组:', np.arange(8).reshape((2, 4)))print('三维数组:', np.arange(60).reshape((3, 4, 5)))print('指定范围三维数组:',np.random.randint(1, 8, size=(3, 4, 5)))
默认一维数组: [0 1 2 3 4]
自定义起点一维数组: [1 2 3 4]
自定义起点步长一维数组: [2 4 6 8]
二维数组: [[01 2 3]
[4 5 6 7]]
三维数组: [[[ 01 2 3 4]
[5 6 7 8 9]
[10 11 12 13 14]
[15 16 17 18 19]]
[[20 21 22 23 24]
[25 26 27 28 29]
[30 31 32 33 34]
[35 36 37 38 39]]
[[40 41 42 43 44]
[45 46 47 48 49]
[50 51 52 53 54]
[55 56 57 58 59]]]
指定范围三维数组: [[[2 3 2 1 5]
[6 5 5 6 7]
[4 4 6 5 3]
[2 2 3 5 6]]
[[2 1 2 4 4]
[1 4 2 1 4]
[4 4 3 4 2]
[4 1 4 4 1]]
[[6 2 2 7 6]
[2 6 1 5 5]
[2 6 7 2 1]
[3 3 1 4 2]]]
[[[3 3 5 6]
[2 1 6 6]
[1 1 3 5]]
[[7 6 5 3]
[5 6 5 4]
[6 5 7 1]]]
shape()用法

查看矩阵或者数组的维数

建立一个3×3的单位矩阵e, e.shape为(3,3)