python numpy 數組如何對每個元素進行操作?

python filter 如何應用在numpy array中,比如我有一個二維numpy數組,想將數組中小於0的置零大於零的保留原值,但是不想用for循環,有沒有比較高效簡潔的方法


答主總結了四種方式,都沒有直接用到循環,希望有用。

&>&>&> import numpy as np
&>&>&> a = np.random.randint(-5, 5, (5, 5))
&>&>&> a
array([[-4, -4, -5, 2, 1],
[-1, -2, -1, 3, 3],
[-1, -2, 3, -5, 3],
[ 0, -3, -5, 1, -4],
[ 0, 3, 1, 3, -4]])
# 方式一
&>&>&> np.maximum(a, 0)
array([[0, 0, 0, 2, 1],
[0, 0, 0, 3, 3],
[0, 0, 3, 0, 3],
[0, 0, 0, 1, 0],
[0, 3, 1, 3, 0]])
# 方式二
&>&>&> (a + abs(a)) / 2
array([[0, 0, 0, 2, 1],
[0, 0, 0, 3, 3],
[0, 0, 3, 0, 3],
[0, 0, 0, 1, 0],
[0, 3, 1, 3, 0]])
# 方式三
&>&>&> b = a.copy()
&>&>&> b[b &< 0] = 0 &>&>&> b
array([[0, 0, 0, 2, 1],
[0, 0, 0, 3, 3],
[0, 0, 3, 0, 3],
[0, 0, 0, 1, 0],
[0, 3, 1, 3, 0]])
# 方式四
&>&>&> np.where(a &> 0, a, 0)
array([[0, 0, 0, 2, 1],
[0, 0, 0, 3, 3],
[0, 0, 3, 0, 3],
[0, 0, 0, 1, 0],
[0, 3, 1, 3, 0]])


可以採用象徵索引,在索引中寫入布爾面具。

如果原數組為a,則 a[a&<0] = 0,就可以實現小於零的值賦值為零。


np.where 一行代碼實現


樓上已經回答了你的問題,我再推薦給你一本書《Python for data analysis》,裡面有一章講numpy的,看一看應該很有用。


設m是numpy的一個array:

你要的結果=(m&>0)*m


恩,用布爾數組可以直接實現啊

import numpy as np

from numpy import random as nr

x = nr.randint(-5,5,(5,5))

y=x.copy()#複製一份出來

booly = y&<0 #小於0的部分是TRUE 大於0是FALSE boolx.shape 等於x.shape

y[booly]=0#y就是你要的數組


這個需求還是挺好實現的,但因為要遍歷,加上我本身不是做Python的,有點外行,所以只會循環。簡單的Demo如下:

#一個二維數組,元素按照題主的要求有正有負。

list1 = [[1,1],[-2,2],[3,-3],[-4,-4]]

#一個空數組,用於接受需求轉換以後的數據。

list2 = []

#for循環,遍曆數組中的元素。

for i ,j in list1 :

#按題主要求,負數變成0,非負數保留自身。樓上說條件索引的方法,大概也是在這一步的操作。

但因為是二維數組,所以想來太抽象,不適合大我這樣思維能力一般的人。所以我用了絕對值求 平均數的笨方法。

[i,j] = [(i+abs(i))/2,(j+abs(j))/2 ]

#把轉換後的數組元素追加到新數組。

list2.append([i,j])

#查看結果,[[1, 1], [0, 2], [3, 0], [0, 0]],複合題主需求。

print list2

一共五行代碼~ 也算簡潔吧。

不知道題主學來Python做什麼,如果和是數據分析的話,推薦Coursera上南京大學的基礎課程。


推薦閱讀:

python 中如何實現一行輸入多個值 ?
為什麼看不懂廖雪峰的Python學習教程?
Python中 pickle有什麼意義,pickle了再恢復?
在Python中如何修改字元串中的某一位字元?
Python3.5.1中如何將input讀取的字元串改為數字?

TAG:Python入門 | numpy |