python如何创建不同元素的矩阵_python创建矩阵

创建二维数组的办法

直接创建（不推荐）

使用模块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）

参考博文：https://blog.csdn.net/kancy110/article/details/69665164

行走菜鸟界的小辣鸡~