【NumPy基础】100道numpy练习——初学与入门篇

时间:2023-02-13 17:36:04

【NumPy基础】100道numpy练习——初学与入门篇

@author:wepon

@blog:http://blog.csdn.net/u012162613/article/details/42784403


今天在deeplearning.net上看theano tutorial,发现一个numpy-100-exercise,介绍numpy一些基本用法的,不过不是很具体,我利用闲暇时间照着敲了一些,权且当作翻译吧,增加函数的原型和详细介绍。持续更新。


一、初学者10道

1、在python环境中导入numpy包,并命名为np
<span style="font-size:18px;">>>> import numpy as np</span>

2、查看numpy版本和配置信息
<span style="font-size:18px;">>>> print np.__version__
>>> np.__config__.show()</span>

3、创建零向量,zeros函数
<span style="font-size:18px;">>>> z=np.zeros((2,3))
>>> print z
[[ 0.  0.  0.]
 [ 0.  0.  0.]]</span>

4、将上面的零向量的第二行第三列元素置为1。注意python中行列下班是从0开始。
<span style="font-size:18px;">>>>z[1,2]=1
>>> print z
[[ 0.  0.  0.]
 [ 0.  0.  1.]]</span>

5、arange函数,创建一个在给定范围的向量。
<span style="font-size:18px;">>>> z=np.arange(1,101)    %1~100范围,注意不包括101
>>> print z</span>

6、reshape函数,将array变形为矩阵
<span style="font-size:18px;">>>> Z = np.arange(9).reshape(3,3)
>>> print Z
[[0 1 2]
 [3 4 5]
 [6 7 8]]</span>

7、nonzero函数,寻找非0元素的下标
<span style="font-size:18px;">>>> nz=np.nonzero([1,2,3,0,0,4,0])
>>> nz
(array([0, 1, 2, 5]),)</span>

8、eye函数,生成单位向量
<span style="font-size:18px;">>>> z=np.eye(3)
>>> print z
[[ 1.  0.  0.]
 [ 0.  1.  0.]
 [ 0.  0.  1.]]</span>

9、diag函数,diagonal对角线。
<span style="font-size:18px;">>>> z=np.diag([1,2,3,4],k=0)   %k=0,以[1,2,3,4]为对角线
>>> print z
[[1 0 0 0]
 [0 2 0 0]
 [0 0 3 0]
 [0 0 0 4]]

>>> z=np.diag([1,2,3,4],k=1)   %k=1,[1,2,3,4]在对角线上一行
>>> print z
[[0 1 0 0 0]
 [0 0 2 0 0]
 [0 0 0 3 0]
 [0 0 0 0 4]
 [0 0 0 0 0]]

>>> z=np.diag([1,2,3,4],k=-1)  %k=-1,[1,2,3,4]在对角线下一行
>>> print z
[[0 0 0 0 0]
 [1 0 0 0 0]
 [0 2 0 0 0]
 [0 0 3 0 0]
 [0 0 0 4 0]]</span>

10、random模块的random函数,生成随机数
<span style="font-size:18px;">>>> Z = np.random.random((3,3))
>>> print Z
[[ 0.95171484  0.61394126  0.38864802]
 [ 0.41943918  0.9398714   0.31608202]
 [ 0.9993507   0.91717093  0.73002723]]</span>



二、入门级10道

1、创建一个8*8的“棋盘”矩阵。
<span style="font-size:18px;">>>> z=np.zeros((8,8),dtype=int)
>>> z[1::2,::2]=1     %1、3、5、7行&&0、2、4、6列的元素置为1
>>> print z
[[0 0 0 0 0 0 0 0]
 [1 0 1 0 1 0 1 0]
 [0 0 0 0 0 0 0 0]
 [1 0 1 0 1 0 1 0]
 [0 0 0 0 0 0 0 0]
 [1 0 1 0 1 0 1 0]
 [0 0 0 0 0 0 0 0]
 [1 0 1 0 1 0 1 0]]
>>> z[::2,1::2]=1
>>> print z
[[0 1 0 1 0 1 0 1]
 [1 0 1 0 1 0 1 0]
 [0 1 0 1 0 1 0 1]
 [1 0 1 0 1 0 1 0]
 [0 1 0 1 0 1 0 1]
 [1 0 1 0 1 0 1 0]
 [0 1 0 1 0 1 0 1]
 [1 0 1 0 1 0 1 0]]</span>

2、min()、max()函数
<span style="font-size:18px;">>>> z=np.random.random((10,10))
>>> zmin,zmax=z.min(),z.max()
>>> print zmin,zmax
0.014230501632 0.99548760299</span>

3、函数tile(A,reps),reps即重复的次数,不仅可以是数字,还可以是array。比如构造棋盘矩阵:
<span style="font-size:18px;">>>> z=np.tile(np.array([[0,1],[0,1]]),(4,4))
>>> print z
[[0 1 0 1 0 1 0 1]
 [0 1 0 1 0 1 0 1]
 [0 1 0 1 0 1 0 1]
 [0 1 0 1 0 1 0 1]
 [0 1 0 1 0 1 0 1]
 [0 1 0 1 0 1 0 1]
 [0 1 0 1 0 1 0 1]
 [0 1 0 1 0 1 0 1]]</span>

4、归一化,将矩阵规格化到0~1,即最小的变成0,最大的变成1,最小与最大之间的等比缩放。

<span style="font-size:18px;">>>> Z = np.random.random((5,5))
>>> Zmax,Zmin = Z.max(), Z.min()
>>> Z = (Z - Zmin)/(Zmax - Zmin)
>>> print Z
[[ 0.          0.32173291  0.17607851  0.6270374   

0.95000808]
 [ 0.49153473  0.70465605  0.61930085  0.00303294  1.        

]
 [ 0.4680561   0.88742782  0.29899683  0.80704789  

0.12300414]
 [ 0.05094248  0.23065875  0.82776775  0.07873239  

0.50644422]
 [ 0.27417053  0.78679222  0.517819    0.5649124   0.4716856 

]]</span>

5、矩阵点乘
<span style="font-size:18px;">>>> z=np.dot(np.ones((5,3)),np.ones((3,2)))
>>> print z
[[ 3.  3.]
 [ 3.  3.]
 [ 3.  3.]
 [ 3.  3.]
 [ 3.  3.]]</span>

6、矩阵相加,5*5矩阵+1*5的向量,相当于每一行都加上1*5矩阵

<span style="font-size:18px;">>>> Z = np.zeros((5,5))
>>> Z += np.arange(5)
>>> print Z
[[ 0.  1.  2.  3.  4.]
 [ 0.  1.  2.  3.  4.]
 [ 0.  1.  2.  3.  4.]
 [ 0.  1.  2.  3.  4.]
 [ 0.  1.  2.  3.  4.]]</span>


7、linspace函数,在给定区间中生成均匀分布的给定个数。
函数原型 linspace(start, stop, num=50, endpoint=True, retstep=False)

<span style="font-size:18px;">>>> Z = np.linspace(0,10,11,endpoint=True, retstep=False)
>>> print Z
[  0.   1.   2.   3.   4.   5.   6.   7.   8.   9.  10.]</span>

生成0~10之间均匀分布的11个数,包括0和1。
若endpoint=False,则10不包括在里面。
若retstep=False,会同时返回均匀区间中每两个数的间隔。

8、sort函数。调用random模块中的random函数生成10个随机数,然后sort排序。

<span style="font-size:18px;">>>> Z = np.random.random(10)
>>> Z.sort()
>>> print Z
[ 0.15978787  0.28050494  0.35865916  0.40047826  0.45141311  

0.4828367
  0.66133575  0.66775779  0.69278544  0.98095989]</span>

9、allclose函数,判断两个array在误差范围内是否相等
函数原型allclose(a, b, rtol=1e-05, atol=1e-08),若absolute(a - b) <= (atol + rtol * absolute(b))则相等。
<span style="font-size:18px;">A = np.random.randint(0,2,5)
B = np.random.randint(0,2,5)
equal = np.allclose(A,B)
print equal</span>

10、mean函数,求平均值

<span style="font-size:18px;">>>> Z = np.random.random(30)
>>> m = Z.mean()
>>> print m
0.362299527973

>>> A = np.random.randint(0,2,5)
>>> B = np.random.randint(0,2,5)
>>> equal = np.allclose(A,B)
>>> print equal
False
</span>

注:randint(min,max,num)生成大小为num的array,数值范围min~max



----------------------------2015/1/15-------------------------------