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ù),稱(chēng)為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的大?。?。
# 用于迭代數(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-09
python中bytes和str類(lèi)型的區(qū)別
這篇文章主要介紹了python中bytes和str類(lèi)型的區(qū)別,文中通過(guò)示例代碼介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2019-10-10
Python?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-06
python 數(shù)字類(lèi)型和字符串類(lèi)型的相互轉(zhuǎn)換實(shí)例
今天小編就為大家分享一篇python 數(shù)字類(lèi)型和字符串類(lèi)型的相互轉(zhuǎn)換實(shí)例,具有很好的參考價(jià)值,希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2018-07-07
Python實(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

