Python NumPy教程之遍歷數(shù)組詳解
NumPy 包包含一個(gè)迭代器對(duì)象numpy.nditer
。它是一個(gè)高效的多維迭代器對(duì)象,使用它可以迭代數(shù)組。使用 Python 的標(biāo)準(zhǔn)迭代器接口訪問(wèn)數(shù)組的每個(gè)元素。
# 用于遍歷數(shù)組的 Python 程序 import numpy as geek # 使用排列方法創(chuàng)建數(shù)組 a = geek.arange(12) # 具有 3 行和 4 列的形狀數(shù)組 a = a.reshape(3,4) print('Original array is:') print(a) print() print('Modified array is:') # 迭代數(shù)組 for x in geek.nditer(a): print(x)
輸出:
Original array is:
[[ 0 1 2 3]
[ 4 5 6 7]
[ 8 9 10 11]]
Modified array is:
0 1 2 3 4 5 6 7 8 9 10 11
選擇迭代的順序以匹配數(shù)組的內(nèi)存布局,而不考慮特定的順序。這可以通過(guò)迭代上述數(shù)組的轉(zhuǎn)置來(lái)看出。
# 用于迭代轉(zhuǎn)置的 Python 程序 # array import numpy as geek # 使用排列方法創(chuàng)建數(shù)組 a = geek.arange(12) # 具有 3 行和 4 列的形狀數(shù)組 a = a.reshape(3,4) print('Original array is:') print(a) print() # 原始數(shù)組的轉(zhuǎn)置 b = a.T print('Modified array is:') for x in geek.nditer(b): print(x)
輸出:
Original array is:
[[ 0 1 2 3]
[ 4 5 6 7]
[ 8 9 10 11]]
Modified array is:
0 1 2 3 4 5 6 7 8 9 10 11
控制迭代順序
有時(shí)以特定順序訪問(wèn)數(shù)組元素很重要,而與內(nèi)存中元素的布局無(wú)關(guān)。nditer 對(duì)象提供了一個(gè) order 參數(shù)來(lái)控制迭代的這一方面。具有上述行為的默認(rèn)設(shè)置是 order='K' 以保持現(xiàn)有順序。這可以用 order='C' 覆蓋 C 訂單和 order='F' 用于 Fortran 訂單。
代碼#1:
# 用于使用特定順序迭代數(shù)組的 Python 程序 import numpy as geek # 使用排列方法創(chuàng)建數(shù)組 a = geek.arange(12) # 具有 3 行和 4 列的形狀數(shù)組 a = a.reshape(3,4) print('Original array is:') print(a) print() print('Modified array in C-style order:') # 具有 3 行和 4 行的形狀數(shù)組,以給定的順序排列數(shù)組 for x in geek.nditer(a, order = 'C'): print(x)
輸出:
Original array is:
[[ 0 1 2 3]
[ 4 5 6 7]
[ 8 9 10 11]]
Modified array in C-style order:
0 1 2 3 4 5 6 7 8 9 10 11
代碼#2:
# 用于使用特定順序迭代數(shù)組的 Python 程序 import numpy as geek # 使用排列方法創(chuàng)建數(shù)組 a = geek.arange(0,60,5) # 具有 3 行和 4 列的形狀數(shù)組 a = a.reshape(3,4) print('Original array is:') print(a) print() print('Modified array in F-style order:') # 以給定順序迭代數(shù)組 for x in geek.nditer(a, order = 'F'): print(x)
輸出:
Original array is:
[[ 0 1 2 3]
[ 4 5 6 7]
[ 8 9 10 11]]
Modified array in F-style order:
0 4 8 1 5 9 2 6 10 3 7 11
修改數(shù)組值
nditer 對(duì)象有另一個(gè)可選參數(shù),稱為op_flags。其默認(rèn)值為只讀,但可以設(shè)置為讀寫(xiě)或只寫(xiě)模式。這將啟用使用此迭代器修改數(shù)組元素。
# 用于修改數(shù)組值的 Python 程序 import numpy as geek # 使用排列方法創(chuàng)建數(shù)組 a = geek.arange(12) # 具有 3 行和 4 列的形狀數(shù)組 a = a.reshape(3,4) print('Original array is:') print(a) print() # 修改數(shù)組值 for x in geek.nditer(a, op_flags = ['readwrite']): x[...] = 5*x print('Modified array is:') print(a)
輸出:
Original array is:
[[ 0 1 2 3]
[ 4 5 6 7]
[ 8 9 10 11]]
Modified array is:
[[ 0 5 10 15]
[20 25 30 35]
[40 45 50 55]]
外環(huán):
類(lèi)nditer
構(gòu)造函數(shù)有一個(gè)flags
參數(shù),可以取以下值
范圍 | 描述 |
---|---|
external_loop | 導(dǎo)致給定的值是具有多個(gè)值的一維數(shù)組,而不是零維數(shù)組 |
c_index | C_order索引可以被跟蹤 |
f_index | 跟蹤 Fortran_order 索引 |
multi-index | 可以跟蹤每次迭代一個(gè)索引的類(lèi)型 |
代碼#1:
# 使用外部循環(huán)迭代數(shù)組值的 Python 程序 import numpy as geek # 使用排列方法創(chuàng)建數(shù)組 a = geek.arange(12) # 具有 3 行和 4 列的形狀數(shù)組 a = a.reshape(3,4) print('Original array is:') print(a) print() print('Modified array is:') for x in geek.nditer(a, flags = ['external_loop'], order = 'C'): print(x)
輸出:
Original array is:
[[ 0 1 2 3]
[ 4 5 6 7]
[ 8 9 10 11]]
Modified array is:
[ 0 1 2 3] [ 4 5 6 7] [8 9 10 11]
代碼#2:
# 使用 f_index 迭代數(shù)組值的 Python 程序 import numpy as geek # 使用排列方法創(chuàng)建數(shù)組 a = geek.arange(6) # 具有 2 行和 3 列的形狀數(shù)組 a = a.reshape(2,3) print('Original array is:') print(a) print() # 使用 f_index 參數(shù)迭代數(shù)組 it = geek.nditer(a, flags=['f_index']) while not it.finished: print("%d <%d>" % (it[0], it.index), end=" ") it.iternext()
輸出:
Original array is:
[[ 0 1 2]
[ 3 4 5]]
0 <0> 1 <2> 2 <4> 3 <1> 4 <3> 5 <5>
廣播迭代
如果兩個(gè)數(shù)組是可廣播的,則組合的nditer對(duì)象能夠同時(shí)對(duì)它們進(jìn)行迭代。假設(shè)一個(gè)數(shù)組a的維度為3X4,并且還有另一個(gè)維度為1X4的數(shù)組b,則使用以下類(lèi)型的迭代器(數(shù)組b廣播到a的大?。?/p>
# 用于迭代數(shù)組的 Python 程序 import numpy as geek # 使用排列方法創(chuàng)建數(shù)組 a = geek.arange(12) # 具有 3 行和 4 列的形狀數(shù)組 a = a.reshape(3,4) print('First array is:') print(a) print() # 使用數(shù)組方法創(chuàng)建第二個(gè)數(shù)組 print('Second array is:') b = geek.array([5, 6, 7, 8], dtype = int) print(b) print() print('Modified array is:') for x,y in geek.nditer([a,b]): print("%d:%d" % (x,y))
輸出:
First array is:
[[ 0 1 2 3]
[ 4 5 6 7]
[ 8 9 10 11]]
Second array is:
[5 6 7 8]
Modified array is:
0:5 1:6 2:7 3:8 4:5 5:6 6:7 7:8 8:5 9:6 10:7 11:8
到此這篇關(guān)于Python NumPy教程之遍歷數(shù)組詳解的文章就介紹到這了,更多相關(guān)Python NumPy遍歷數(shù)組內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家!
相關(guān)文章
python基于FTP實(shí)現(xiàn)文件傳輸相關(guān)功能代碼實(shí)例
這篇文章主要介紹了python基于FTP實(shí)現(xiàn)文件傳輸相關(guān)功能代碼實(shí)例,文中通過(guò)示例代碼介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2019-09-09python中bytes和str類(lèi)型的區(qū)別
這篇文章主要介紹了python中bytes和str類(lèi)型的區(qū)別,文中通過(guò)示例代碼介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2019-10-10Python?Scrapy庫(kù)構(gòu)建基礎(chǔ)爬蟲(chóng)
這篇文章主要為大家介紹了Python?Scrapy庫(kù)構(gòu)建基礎(chǔ)爬蟲(chóng)示例詳解,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進(jìn)步,早日升職加薪2023-08-08對(duì)Python的交互模式和直接運(yùn)行.py文件的區(qū)別詳解
今天小編就為大家分享一篇對(duì)Python的交互模式和直接運(yùn)行.py文件的區(qū)別詳解,具有很好的參考價(jià)值,希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2019-06-06python 數(shù)字類(lèi)型和字符串類(lèi)型的相互轉(zhuǎn)換實(shí)例
今天小編就為大家分享一篇python 數(shù)字類(lèi)型和字符串類(lèi)型的相互轉(zhuǎn)換實(shí)例,具有很好的參考價(jià)值,希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2018-07-07Python實(shí)現(xiàn)簡(jiǎn)單生成驗(yàn)證碼功能【基于random模塊】
這篇文章主要介紹了Python實(shí)現(xiàn)簡(jiǎn)單生成驗(yàn)證碼功能,結(jié)合實(shí)例形式分析了Python基于random模塊生成隨機(jī)字符串的相關(guān)操作技巧,需要的朋友可以參考下2018-02-02