結合numpy陣列和掩模

Question

我有以下問題。在知道與numpy我可以閱讀genfromtext數據文件。然後我可以用掩碼（從字典中）選擇記錄。

>>> gln = a[a['AA2'] == 'GLN'] 
>>> gln['mean'].mean() 
62.805000305175781 

我怎樣才能做到這一點，而不是從磁盤讀取這些數組的組合呢？

x=[] 
y=[] 
ov=[] 
aa=[] 

a = numpy.array(x,y,ov,aa)   <== does not actually work 
gln = a[a[3] == 'GLN'] 
gln['mean'].mean() 

我如何得到這個工作？

Answer 1

請記住，在Python中，由[]生成的對象通常被稱爲list，而不是array。所以考慮到這一點，我假設你真正的問題是「如何將4個獨立的python list s轉換爲2維numpy陣列？」

答案取決於您希望list s如何排列在陣列中。假設每個列表代表一排，你可以這樣做：

>>> x = range(0, 4) 
>>> y = range(4, 8) 
>>> ov = range(8, 12) 
>>> aa = range(12, 16) 
>>> numpy.array([x, y, ov, aa]) 
array([[ 0, 1, 2, 3], 
     [ 4, 5, 6, 7], 
     [ 8, 9, 10, 11], 
     [12, 13, 14, 15]]) 

或者，如果每個列表代表一欄，你可以這樣做：

>>> numpy.array(zip(x, y, ov, aa)) 
array([[ 0, 4, 8, 12], 
     [ 1, 5, 9, 13], 
     [ 2, 6, 10, 14], 
     [ 3, 7, 11, 15]]) 

然後，你可以只使用布爾索引你與genfromtxt產生的array做：

>>> a = numpy.array(zip(x, y, ov, aa)) 
>>> a[a == 1] 
array([ 1, 5, 9, 13, 3, 7, 11, 15]) 

這可能是你正在尋找合併x的一些其他的方式，y,ov和aa組成一個數組。在這種情況下，您將不得不指定列表中的內容和所需輸出的形狀。