如何对角排列数组中的元素

时间:2015-11-19 20:03:44

标签: python arrays diagonal

使用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.]]

2 个答案:

答案 0 :(得分: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]])

但是,如果你想进行转换/旋转,那么你必须确定你希望如何发生这种情况(我不知道你将如何在非方形矩阵上工作)。

答案 1 :(得分:0)

请注意,您正在尝试将值设置为对角线形式,因此您将要将下一个-6位置放在右侧,而不是前一个位置。由于你每次都要将某个东西索引为一个,这听起来像是一个循环的好地方。