的Python：在numpy的陣列高效運營

Question

說我有一個numpy的數組x：

x = array([[ 3, 2, 1], 
      [ 3, 25, 34], 
      [ 33, 333, 3], 
      [ 43, 32, 2]]) 

我要執行以下操作沒有明確地寫一個for循環即表示它使用內置的自動方法循環;

1）的所有1即

x = array([[ 3, 1, 1], 
      [ 3, 1, 34], 
      [ 33, 1, 3], 
      [ 43, 1, 2]]) 

2）在原始陣列的列替換第二列中，替換與第二和第三的產物第三列即

x = array([[ 3, 2, 1*2], 
      [ 3, 25, 34*25], 
      [ 33, 333, 3*333], 
      [ 43, 32, 2*32]]) 

3 ）最後，我想根據條件替換原始陣列中的第二列，例如

x[1] = 0 if x[0] > 5 else 4 

即陣列現在看起來像：

x = array([[ 3, 4, 1], 
      [ 3, 4, 34], 
      [ 33, 0, 3], 
      [ 43, 0, 2]]) 

有什麼建議嗎？ 謝謝！

Answer 1

上numpy的文檔是很值得一讀，因爲這是很基本的東西...

1. x[:,1] = 1
2. x[:,2] *= x[:,1]
3. x[:,1] = np.where(x[:,0] > 5, 0, 4)