在numpy陣列中，我想複製一列到另一個矩陣，並從原始矩陣中刪除列

Question

mat=[[0,1,5],[1,3,6],[-1,4,4],[1,2,2],[7,3,7],[2,5,3]] 

mat矩陣形狀可能是10000 * 5。這裏只是一個例子

這裏我定義一個函數。它試圖找到mat [：，0] < be或mat [：，0]> ba或mat [：，1] < bb。如果一列與條件匹配，則元素[i，0] <爲元素[i，0] = be，然後將該列複製到另一個矩陣「swape」。還要從矩陣「mat」中刪除這一列。與mat [：，0]相同> ba或mat [：，1] < bb。對於mat [：，1] < bb，colume將複製到「swapt」，mat [：，0]> ba不復制，只需刪除。

例如是= 0，BA = 6，BB = 3

返回列表應該是：

mat=[[1,3,6],[2,5,3]] 
swape=[[-1,4,4]] 
swapt=[[1,2,2],[0,1,5]] 

該函數將返回墊，swape和SWAPT

def swapco(be,ba,bb,mat): 
     swape=np.array() 
     swapt=np.array() 
     leng=np.shape(mat)[0] 
     for i in range(leng): 
      if mat[i,2]<bb: 
       mat[i,2]=bb 
       np.append(swapt,i,1) 
       np.delete(mat, i, 0) 
      else: 
       if mat[i,0]>=ba:  
        mat[i,0]=ba 
        np.append(swape,i,1) 
        np.delete(mat, i, 0) 
       elif mat[i,0]<=be: 
        mat[i,0]=be 
        np.append(swape,i,1) 
        np.delete(mat, i, 0) 
      i+=1 

     return swape, swapt 

在我的代碼中，我發現一旦某些列匹配條件，矩陣墊的長度總是減小。它會追加和刪除錯誤的列。另外追加是追加一個地址或deepcopy？

如果使用

for col in mat: 

那麼如何刪除自己在墊子？ 或者任何有效的方法來寫這個代碼？

問題更新中...

Answer 1

這是從你的代碼你想要做什麼，但讓我爲你寫的一個小例子，可以幫助您開始清理你的代碼真的不清楚。

import numpy as np 

def split(mat, a, b): 
    assert a < b 
    where_less_than_a = mat[:, 0] < a 
    where_less_than_b = mat[:, 0] < b 

    less_than_a = mat[where_less_than_a, :] 
    between_a_b = mat[(~where_less_than_a) & where_less_than_b, :] 
    greater_eq_b = mat[(~where_less_than_b), :] 
    return less_than_a, between_a_b, greater_eq_b 

mat = np.arange(27).reshape((9, 3)) 
x, y, z = split(mat, 4., 17.) 
print(x) 
# [[0 1 2] 
# [3 4 5]] 
print(y) 
# [[ 6 7 8] 
# [ 9 10 11] 
# [12 13 14] 
# [15 16 17]] 
print(z) 
# [[18 19 20] 
# [21 22 23] 
# [24 25 26]] 

我希望上面的例子可以幫助你開始你自己的項目。有一件事可能有點令人困惑，上面代碼中的~被用作「邏輯非」操作符。它可以以這種方式與bool類型的numpy數組一起使用，但要小心，因爲它對於其他python對象（它被稱爲補數運算符）具有不同的含義。