我有一个以下格式的文本文件:
123, 456, [45,45,90]
20, 19, [4,67,95]
120, 45, [50,20,20]
我想知道最好的方法是为每个逗号分隔值分配一个变量。例如,
x, y, vec = [], [], []
for i, line in enumerate(file.readlines()):
x.append(float(line.split(',')[0]))
y.append(float(line.split(',')[1]))
vec.append(np.array(line.split(',')[2]))
>>> IDEAL OUTPUT
x = [123,20,120]
y = [456, 19, 45]
vec = [[45,45,90],[4,67,95],[50,20,20]]
对于变量vec
,我应该对上面的代码进行哪些编辑,以使变量vec
成为一个numpy数组?
感谢您的帮助。