python 将数组中部分转入新数组_python – 如何将掩码从一个数组应用到另一个数组?...
現(xiàn)在我已經(jīng)讀過了masked的數(shù)組文檔,搜索到處都是愚蠢的。我無法想到我的生活中如何將一個面具從一個陣列應(yīng)用到另一個陣列。
例:
import numpy as np
y = np.array([2,1,5,2]) # y axis
x = np.array([1,2,3,4]) # x axis
m = np.ma.masked_where(y>2, y) # filter out values larger than 5
print m
[2 1 -- 2]
print np.ma.compressed(m)
[2 1 2]
所以這樣做很好….但是要繪制這個y軸,我需要一個匹配的x軸。如何將y數(shù)組中的蒙版應(yīng)用于x數(shù)組?這樣的事情是有道理的,但產(chǎn)生垃圾:
new_x = x[m.mask].copy()
new_x
array([5])
那么,如何做到這一點(注意新的x數(shù)組需要是一個新的數(shù)組)。
編輯:
好的,這樣做的一種方法是這樣工作的:
>>> import numpy as np
>>> x = np.array([1,2,3,4])
>>> y = np.array([2,1,5,2])
>>> m = np.ma.masked_where(y>2, y)
>>> new_x = np.ma.masked_array(x, m.mask)
>>> print np.ma.compressed(new_x)
[1 2 4]
但這是令人難以置信的凌亂!我試圖找到像IDL一樣優(yōu)雅的解決方案…
總結(jié)
以上是生活随笔為你收集整理的python 将数组中部分转入新数组_python – 如何将掩码从一个数组应用到另一个数组?...的全部內(nèi)容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 妄想山海草龟在哪里?
- 下一篇: 5m 云服务器2核4g_华为云服务器2核