2015-11-19 95 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]]) 

但是,如果你想做一个转化/旋转,那么你必须确定你想怎么这样的事情发生(我不知道你是如何工作的是在非方形矩阵上)。