类型错误：列表索引必须是整数或切片，而不是在 python 中使用 sys 导入的元组

Question

在读入数据时，Python 给了我以下错误：

类型错误：列表索引必须是整数或切片，而不是在 python 中使用 sys 导入的元组

我使用以下 data.txt 输入文件：

1.0 2.0 3.0
4.0 5.0 6.0

使用命令 !python file_name data.txt import sys

fp = open(sys.argv[1],"r+")
coordinates = fp.readlines()
fp.close()

import numpy

a = numpy.array(coordinates[0, 1, 2])
b = numpy.array(coordinates[3, 4, 5])

dist_unround=numpy.linalg.norm(a-b)
dist_round=round(dist_unround, 2)

energy_unround=0.5*2*((dist_unround-3.0)**2)
energy_round=round(energy_unround,2)

print("dist \t energy")
print(dist_round,"\t" ,energy_round) 

我尝试从随机输入文本文件中计算向量。 我的 numpy.array 代码有问题吗？

Answer 1

坐标是一个列表，并且您正试图在a=行上传递元组0, 1 ,2

要正确使用方法，请更换

a = numpy.array(coordinates[0, 1, 2])
b = numpy.array(coordinates[3, 4, 5])

和

a = numpy.array(coordinates[0].split())
b = numpy.array(coordinates[1].split())

请注意， readlines返回一个字符串列表。 调用该列表（ coordinates[0] ）上的索引 0 以获取第一组（行）值，然后对该字符串使用.split()方法使它们成为由空格分隔的字符串列表。

您可能希望在使用时将它们转换为浮点数或整数。 如果你愿意，可以用列表理解来做到这一点

a = numpy.array([int(x) for x in coordinates[0].split()])
b = numpy.array([int(x) for x in coordinates[1].split()])

Answer 2

从你提供的小细节来看，我猜这就是你要找的——

arr=[]
with open("data.txt") as f:
    for line in f:
        a=list(line.split(" "))
        for element in a:
            arr.append(float(element.rstrip()))
print(arr)