0

我有一个数组,我正在对其进行一些计算。数组开始为

e = array([[1,2,1,3],
           [3,-1,-3,-1],
           [2,3,1,4]])

我对其进行了一些修改以将其转换为:

array([[  1,   2,   1,   3],
      [  0,  -7,  -6, -10],
      [  0,  -1,  -1,  -2]])

然后我在上面运行这段代码:

import numpy as np
from fractions import Fraction

def ref(x):
    dimension = x.shape
    row_counter = 1
    first_values = [x[i][row_counter] for i in range(dimension[0])]                               #gets a list of elements of the column
    first_values = [number != 0 for number in first_values]                                       #0 is a pivot element?
    if False in first_values:                                                                     
        true_index =  first_values.index(True); false_index =   first_values.index(False)
        if true_index > false_index:                                                               #not any more
            x[[false_index, true_index]] = x[[true_index, false_index]]

    for i in range(row_counter+1,dimension[0]):
        multiplier = Fraction(x[row_counter][row_counter], x[i][row_counter])**-1
        row1 = multiplier*x[row_counter]
        row1 = x[i]-row1
        print row1
        x[i] = row1   

return x

运行此返回:

[0 0 -1/7 -4/7]
array([[  1,   2,   1,   3],
         [  0,  -7,  -6, -10],
         [  0,   0,   0,   0]])

所以结果应该是

array([[  1,   2,   1,   3],
         [  0,  -7,  -6, -10],
         [  0, 0, -1/7, -4/7]])

它打印正确的行条目,但不会添加到数组中,而是添加一行零。有人可以告诉我为什么吗?谢谢。

4

1 回答 1

2

通常,numpy数组与特定类型是同构的。例如:

>>> a = np.array([1,2,3])
>>> a
array([1, 2, 3])
>>> a.dtype
dtype('int64')

当您专门设置元素或切片时,您添加的内容会被强制转换为 current dtype,因此:

>>> a[0] = 5
>>> a
array([5, 2, 3])

>>> a[0] = 4.3
>>> a
array([4, 2, 3])

当你没有就地行动时,你可以向上转型,因此numpy无论如何都必须制作一个副本(即一个新对象):

>>> a = np.array([1,2,3])
>>> a + 4.3
array([ 5.3,  6.3,  7.3])
>>> (a + 4.3).dtype
dtype('float64')

numpy在您的情况下,如果您从dtype 数组开始,您可以获得所需的行为object

>>> e = np.array([[  1,   2,   1,   3],
...               [  0,  -7,  -6, -10],
...               [  0,  -1,  -1,  -2]], dtype=object)
>>> 
>>> ref(e)
array([[1, 2, 1, 3],
       [0, -7, -6, -10],
       [0, 0, -1/7, -4/7]], dtype=object)
于 2013-01-17T02:52:05.680 回答