python - 使用 python 和 numpy 解析文件中的坐标

Question

我有一个包含 MD 模拟包生成的坐标的文件。

格式如下:

(1.323232、1.22323、3.23123)

您还可以按如下方式指定坐标的输出，不带逗号:

(1.323232 1.22323 3.23123)

如何使用 python 解析数组中的这样一行。具体如何删除括号。在 C 语言中，使用 scanf 很容易做到这一点？

Answer 1

有两种明显的方法可以做到这一点:使用或不使用正则表达式。由于您可能会得到 300 个正则表达式答案，因此让我们展示如何在没有正则表达式的情况下做到这一点。

我们想要去掉括号，然后按空格或逗号后跟空格进行分割。另一种说法是:去掉括号，然后按空格分割，然后去掉可选的尾随逗号。例如:

line = line[1:-1] # strip the parens
bits = line.split() # split on whitespace
bits = [bit.rstrip(',') for bit in bits] # strip trailing commas
bits = map(float, bits) # convert to float

当然，您可以将所有这些合并到一行中:

bits = [float(bit.rstrip(',')) for bit in line[1:-1].split()]