如何使用python / sage创建显示矩阵中的列条目计数的行？

大卫·里士满

所以这是我的矩阵：

A = matrix([[1,1,2,2,2], 
            [3,1,4,3,3], 
            [0,4,2,4,0], 
            [0,0,2,0,0], 
            [0,0,2,0,0]])

我想插入一个0行，以显示列条目的数量。

这是我的原始代码：

A = matrix([[1,1,2,2,2],
            [3,1,4,3,3],
            [0,4,2,4,0],
            [0,0,2,0,0],
            [0,0,2,0,0]])
puzzle = A.insert_row(0, sum(A))

这段代码的缺点是，当我只需要每列的计数时，它为我提供了每列的总和。

是）我有的

[ 4  6 12  9  5]
[ 1  1  2  2  2]
[ 3  1  4  3  3]
[ 0  4  2  4  0]
[ 0  0  2  0  0]
[ 0  0  2  0  0]

期望的

[ 2  3  5  3  2]
[ 1  1  2  2  2]
[ 3  1  4  3  3]
[ 0  4  2  4  0]
[ 0  0  2  0  0]
[ 0  0  2  0  0]

最好的，

大卫

约翰·帕尔米里（John Palmieri）

尝试nonzero_positions_in_column方法：

sage: A = matrix([[1,1,2,2,2], 
....:             [3,1,4,3,3], 
....:             [0,4,2,4,0], 
....:             [0,0,2,0,0], 
....:             [0,0,2,0,0]])
sage: A.column(1)
(1, 1, 4, 0, 0)
sage: A.nonzero_positions_in_column(1)
[0, 1, 2]

sage: v = vector([len(A.nonzero_positions_in_column(i)) for i in range(A.ncols())])
sage: A.insert_row(0, v)
[2 3 5 3 2]
[1 1 2 2 2]
[3 1 4 3 3]
[0 4 2 4 0]
[0 0 2 0 0]
[0 0 2 0 0]

本文收集自互联网，转载请注明来源。

如有侵权，请联系 [email protected] 删除。