【每天幾分鐘,從零入門python編程的世界!】
上節我們學習了如何利用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)生成的數組就是[0 1 2]。
所以技巧3的h[np.arange(3),i]這行代碼,
表示從h中取元素,
np.arange(3)表示依次從第1行、第2行、第3行取元素,
i是[2 1 0],代表依次從第3列、第2列、第1列取元素,
所以就是依次從h的第1行第3列、第2行第2列、第3行第1列取元素。
取出結果就是[3 5 7]
★技巧4,批量改變數組的元素:
運行結果:
以上就是numpy中數組查找元素和改變元素的小技巧。
我是時問新,歡迎關注我。跟我一起從零開始學習Python,每天花一點時間,開啟python編程新世界的大門,領略新的風光,讓人生多一種可能!