【python学习笔记】Numpy 例题

Numpy 例题

1. 生成一个一维数组，起始值为5，终点值为15，样本数为10个

   import numpy as np
   a=np.arange(5,15)
   #a=np.array([5,6,7,8,9,10,11,12,13,14])
   print(a)
   

   [ 5 6 7 8 9 10 11 12 13 14]

2. 输出对角矩阵

   #第一种方法
   import numpy as np
   a=np.zeros((3,3))
   for i in range(3):
       for j in range(3):
           if(i==j):
               a[i][i]=1.0
           elif(i!=j):
               a[i][j]=0.0
   print(a)
   

   [[ 1. 0. 0.]

   [ 0. 1. 0.]

   [ 0. 0. 1.]]