python - 将元组的字符串数组转换为输入和输出

Question

有一个这种格式的字符串数组:

stringList = ['(0.0,[0.0,0.0,0.0])\n' , '(1.0,[1.0,0.0,0.0])\n']

如何转换为:

inputs = [0.0 , 1.0]
outputs = [ [0.0,0.0,0.0] , [1.0,0.0,0.0] ]

我最接近的是:

inputs = []
outputs = []

for l in stringList : 
    inputs.append(np.array(float(l.split(',')[0].replace('(' , ''))))

for l in stringList : 
     outputs.append(float(l.split(',')[1].replace('[' , '')))

print(inputs)
print(outputs)

打印:

[array(0.0), array(1.0)]
[0.0, 1.0]

如何转换为:

inputs = [0.0 , 1.0]
outputs = [ [0.0,0.0,0.0] , [1.0,0.0,0.0] ]

?

Answer 1

您的字符串是元组格式。您可以使用 ast.literal_eval 将其计算为一个元组，然后将结果附加到相应的输出中:

stringList = ['(0.0,[0.0,0.0,0.0])\n' , '(1.0,[1.0,0.0,0.0])\n']
inputs = []
outputs = []

from ast import literal_eval
for s in stringList:
    t = literal_eval(s)
    inputs.append(t[0])
    outputs.append(t[1])

inputs
# [0.0, 1.0]

outputs
# [[0.0, 0.0, 0.0], [1.0, 0.0, 0.0]]

或者更短的版本，你可以在literal_eval之后使用zip:

inputs, outputs = zip(*map(literal_eval, stringList))