python 浮点除法不起作用

Question

我有一个Python代码，我想要对矩阵中的两个元素进行浮点除法。我尝试了 float() 和 from future import division 但它们都不起作用。这是我的代码。

from __future__ import division
import numpy as np

def backsub(U, b):
    N = np.shape(U)[0] 
    x = b.copy()
    x[N - 1, 0] = x[N - 1, 0] / U[N-1, N-1]  #This keeps giving me integer results. No matter I do float(x[N - 1, 0]) or from __future__ import division
    print x[N - 1, 0]
    for i in range(N - 2, -1, -1): 
        for j in range(i + 1, N, 1):
            x[i, 0] = x[i, 0] - U[i, j] * x[j]
        x[i, 0] = x[i, 0] / U[i, i]
    return x

b = np.matrix('1; 2; 3')
U = np.matrix('6, 5, 1; 0, 1, 7; 0, 0, 2')
print backsub(U, b)

输出:

1
[[ 4]
 [-5]
 [ 1]]

Answer 1

尝试

x[i, 0] = float(x[i, 0]) / U[i, i]

尝试一下

b = np.matrix('1; 2; 3', dtype=float)
U = np.matrix('6, 5, 1; 0, 1, 7; 0, 0, 2', dtype=float)

或者你也可以尝试

b = np.matrix([[1], [2], [3], dtype=float)
U = np.matrix([[6, 5, 1], [0, 1, 7], [0, 0, 2]], dtype=float)

希望对你有帮助。