2015-11-19 96 views
0

使用Python非常新,但我設法提供了一個代碼,它給了我下面的數組。不會涉及任務的細節,但下一步,我想重新排列-6元素,使其對角放置。如何重新排列陣列中的元素對角線

[[-6. 0. 1. 0. 1. 0. 1.] 
[-6. 0. 1. 0. 1. 1. 0.] 
[-6. 0. 1. 1. 0. 0. 1.] 
[-6. 0. 1. 1. 0. 1. 0.] 
[-6. 1. 0. 0. 1. 0. 1.] 
[-6. 1. 0. 0. 1. 1. 0.] 
[-6. 1. 0. 1. 0. 0. 1.] 
[-6. 1. 0. 1. 0. 1. 0.]] 
+1

舉例說明預期的輸入以及期望的輸出以及迄今爲止解決該問題的代碼。 – shuttle87

+0

我試過用這樣的代碼使用np.argmin函數:b = a [:,np.argmin(a,axis = 1)]。 –

+0

你在找什麼樣的確切轉換?請嘗試用精確的數學術語解釋。 – shuttle87

回答

0

請注意,您正在嘗試將值設置爲對角線形式,因此您將在下一個-6的位置比前一個位置靠右。既然你每次都會索引一個索引,這聽起來像是一個循環的好地方。

1

設置對角線很簡單:

import numpy as np 

a = np.random.randint(0,10,(11,10)) # I notice your matrix is not square! 

In [31] a 
Out[31]: 
array([[0, 7, 6, 0, 8, 6, 4, 1, 0, 9], 
     [4, 7, 6, 0, 3, 4, 6, 6, 9, 2], 
     [2, 3, 9, 6, 2, 3, 0, 0, 2, 7], 
     [5, 9, 6, 8, 3, 3, 0, 8, 3, 5], 
     [0, 0, 7, 0, 4, 8, 2, 4, 8, 7], 
     [3, 3, 1, 6, 6, 4, 9, 5, 1, 2], 
     [8, 1, 8, 4, 9, 6, 6, 4, 2, 7], 
     [0, 8, 0, 1, 1, 9, 9, 7, 7, 1], 
     [9, 2, 6, 2, 8, 9, 4, 8, 0, 8], 
     [4, 4, 2, 7, 7, 4, 4, 7, 8, 6], 
     [5, 3, 5, 8, 1, 7, 3, 5, 5, 7]]) 




np.fill_diagonal(a,a[:,0]) 

In [33] a 
Out[33]: 
array([[0, 7, 6, 0, 8, 6, 4, 1, 0, 9], 
     [4, 4, 6, 0, 3, 4, 6, 6, 9, 2], 
     [2, 3, 2, 6, 2, 3, 0, 0, 2, 7], 
     [5, 9, 6, 5, 3, 3, 0, 8, 3, 5], 
     [0, 0, 7, 0, 0, 8, 2, 4, 8, 7], 
     [3, 3, 1, 6, 6, 3, 9, 5, 1, 2], 
     [8, 1, 8, 4, 9, 6, 8, 4, 2, 7], 
     [0, 8, 0, 1, 1, 9, 9, 0, 7, 1], 
     [9, 2, 6, 2, 8, 9, 4, 8, 9, 8], 
     [4, 4, 2, 7, 7, 4, 4, 7, 8, 4], 
     [5, 3, 5, 8, 1, 7, 3, 5, 5, 7]]) 

但是,如果你想做一個轉化/旋轉,那麼你必須確定你想怎麼這樣的事情發生(我不知道你是如何工作的是在非方形矩陣上)。