Python学习第112课numpy

上节我们学习了如何利用index找到ndarray数组中的一些元素,并把找到的元素生成一个新的ndarray。代码如下:

现在我们学习几个用index找到ndarray中元素的小技巧。

★技巧1:

★技巧2:

以上两种写法运行结果都是:

以上两种简写技巧,相当于从h中找到元素时,通过两个list[0,1,2]和[2,1,0],每个list的元素一一对应,分别取出第1行第3列、第2行第2列、第3行第1列的元素,三个元素分别为3、5、7。

相当于h[]中的第一个list是代表元素所在行,第二个list代表元素所在列。

★技巧3:

运行结果:

说明:

●arange()函数

语法:np.arange([start,]stop,[step,]dtype=None)

作用:生成在numpy中创建等差数组。

参数:

start是起始值,可以不写,不写默认从0开始。

stop是结束值,必须要写,生成的数组的元素不包括结束值

step是步长,可以不写,不写默认步长是1。

numpy的arange()函数和Python的range()函数类似。

因此np.arange(3)生成的数组就是[]。

所以技巧3的h[np.arange(3),i]这行代码,

表示从h中取元素,

np.arange(3)表示依次从第1行、第2行、第3行取元素,

i是[],代表依次从第3列、第2列、第1列取元素,

所以就是依次从h的第1行第3列、第2行第2列、第3行第1列取元素。

取出结果就是[]

★技巧4,批量改变数组的元素:

运行结果:

以上就是numpy中数组查找元素和改变元素的小技巧。

我是时问新,欢迎


转载请注明:http://www.aierlanlan.com/cyrz/2895.html

  • 上一篇文章:
  •   
  • 下一篇文章: