上节我们学习了如何利用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中数组查找元素和改变元素的小技巧。
我是时问新,欢迎