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

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

【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-------------------------------