這篇文章將為大家詳細(xì)講解有關(guān)numpy.where()函數(shù)怎么用,小編覺(jué)得挺實(shí)用的,因此分享給大家做個(gè)參考,希望大家閱讀完這篇文章后可以有所收獲。
近日用到numpy.where()函數(shù),大部分使用方式都能理解,但是在看
>>> x = np.arange(9.).reshape(3, 3) >>> np.where( x > 5 ) (array([2, 2, 2]), array([0, 1, 2]))
的時(shí)候就懵逼了,為什么是兩個(gè)數(shù)組?
后來(lái)才發(fā)現(xiàn)返回值給的其實(shí)是坐標(biāo),以上式為例,返回值其實(shí)是
x[2, 0], x[2, 1], x[2, 2]
即對(duì)于該數(shù)組x,返回值為兩個(gè)數(shù)組a、b,其中滿(mǎn)足條件的值在x中的位置為
x[a[0], b[0]], x[a[1], b[1]], ..., x[a[i], b[i]]
關(guān)于“numpy.where()函數(shù)怎么用”這篇文章就分享到這里了,希望以上內(nèi)容可以對(duì)大家有一定的幫助,使各位可以學(xué)到更多知識(shí),如果覺(jué)得文章不錯(cuò),請(qǐng)把它分享出去讓更多的人看到。