python - TypeError: __getitem__() takes exactly 2 arguments (2 given) TypeError? ( python 3)

Question

我定义了以下函数:

def eigval(matrix):
    a = matrix[0, 0]
    b = matrix[0, 1]
    c = matrix[1, 0]
    d = matrix[1, 1]
    c1 = (a + d) + sqrt((4 * b * c) + ((a - d)**2))
    c2 = (a + d) - sqrt((4 * b * c) + ((a - d)**2))
    return c1 / 2, c2 / 2

创建它是为了找到 2 X 2 矩阵的特征值。我用它在矩阵上迭代运行 Jacobi 算法。传入的矩阵是一个字典，以元组为键表示位置，以 float 为值。这个函数将在大约 6 次迭代中正常工作，但随后我会得到:

TypeError: __getitem__() takes exactly 2 arguments (2 given) 

在 block 的第一行(带有 a 的行)。

我对此完全感到困惑，因为正如我所说，它可以正常运行大约 6 次然后停止。

编辑:这是一个创建我要传入的矩阵类型的函数:(鉴于每次迭代的矩阵都会不同)

def create():
    matrix = {}
    matrix[0, 0] = 2
    matrix[0, 1] = 1
    matrix[1, 0] = 1
    matrix[1, 1] = 2
    return matrix

非常感谢任何帮助! (P.S. 第一篇文章在这里)

Answer 1

你的矩阵是一个使用元组作为键的字典，这可能不是你想要做的。

尝试使用嵌套列表:

from math import sqrt

def create():
    matrix = [[1, 2], [1, 2]]
    return matrix

def eigval(matrix):
    a = matrix[0][0]
    b = matrix[1][0]
    c = matrix[0][1]
    d = matrix[1][1]
    c1 = (a + d) + sqrt((4 * b * c) + ((a - d)**2))
    c2 = (a + d) - sqrt((4 * b * c) + ((a - d)**2))
    return [c1 / 2, c2 / 2]

>>> m = create() 
>>> eigval(m)
[3.0, 0.0]

或者使用 numpy:

import numpy

def eigval(matrix):
    a = matrix[0, 0]
    b = matrix[1, 0]
    c = matrix[0, 1]
    d = matrix[1, 1]
    c1 = (a + d) + sqrt((4 * b * c) + ((a - d)**2))
    c2 = (a + d) - sqrt((4 * b * c) + ((a - d)**2))
    return numpy.array([c1 / 2, c2 / 2])

>>> m = numpy.array([[1, 2], [1, 2]])
>>> m
array([[1, 2],
       [1, 2]])
>>> eigvalues, eigvectors = numpy.linalg.eig(m)
>>> eigvalues
array([ 0.,  3.])

>>> eigval(m)
array([ 3.,  0.])