正在尝试将字符串转换为复数列表

Question

我正在尝试将字符串转换为复数列表。（如果您阅读它时不带引号，那么它将是一个复数列表。）我已经编写了一个函数来执行此操作，但我得到了以下错误：

Traceback (most recent call last):
    File "complex.py", line 26, in <module>
        print(listCmplx('[1.111 + 2.222j, 3.333 + 4.444j]'))
    File "complex.py", line 10, in listCmplx
        while (not isDigit(listIn[count])) and (listIn[count] != '.'):
IndexError: string index out of range

我做错了什么

def isDigit(char):
    return char in '0123456789'

def listCmplx(listIn):
    listOut = []
    count = 0
    real = '0'
    imag = '0'
    while count < len(listIn):
        while (not isDigit(listIn[count])) and (listIn[count] != '.'):
            count += 1
        start = count
        while (isDigit(listIn[count])) or (listIn[count] == '.'):
            count += 1
        end = count
        if listIn[count] == 'j':
            imag = listIn[start:end]
        else:
            real = listIn[start:end]
        if listIn[count] == ',':
            listOut += [float(real) + float(imag) * 1j]
            real = '0'
            imag = '0'
    return listOut

print(listCmplx('[1.111 + 2.222j, 3.333 + 4.444j]'))

先谢谢你

Answer 1

令人惊讶的是，这是Python不需要编写任何函数就可以做到的，它内置了复数类

listIn = '1.111 + 2.222j, 3.333 + 4.444j'
listOut =  eval(listIn)

print(listOut[0])
print(listOut[0].imag,listOut[0].real)