导入结构化数据到蟒蛇

Question

我有一组在其阵列看起来像这样的文本文件：

[(0,1,3),(0,4,5),...(1,9,0)] 

[(9,8,7),(0,4,5),...(1,9,0)] 

其中的行是不一样的长度。

这实质上的路径列表，其中每个点的集合是一个路径，即：

(0,1,3),(0,4,5),...(1,9,0) =path1 

(9,8,7),(0,4,5),...(1,9,0) =path2 

我需要的形式，我可以调用所有元素导入此。例如，对于路径1中的所有点，确定到路径2中所有点的距离。不知道从哪里开始考虑分隔符不想同时使用括号和逗号，然后以可调用的方式构建数组。

Answer 1

以下代码中（假定每行一个路径，并且没有额外的空格）插入numpy的阵列的列表读取数据，然后演示如何计算两个点之间的距离。

import numpy as np 
import numpy.linalg as la 

#replace with your datafile 
datafile = "../data/point_path.txt" 
paths = [] 
with open(datafile, "r") as f: 
    for line in f: 
    point_strs = line.strip().strip("[()]").split("),(") 
    npoints = len(point_strs) 
    path = np.empty((npoints, 3)) 
    for i in xrange(npoints): 
     path[i,:] = np.array(map(int, point_strs[i].split(","))) 
    paths.append(path) 

print "First point of path 1:" 
print paths[0][0] 
print "Second point of path 2:" 
print paths[1][1] 

print "Euclidean Distance between these points:" 
print la.norm(paths[0][0]-paths[1][1]) 

的这个输出是：

First point of path 1: 
[ 0. 1. 3.] 
Second point of path 2: 
[ 0. 4. 5.] 
Euclidean Distance between these points: 
3.60555127546 

编辑：如何格式化输入文件
代码假设点的每个列表是在其自己的行（例如，在F线，解析点的列表）。因此，以下文件：

[(0,2,3),(0,4,0)] [(1,4,5),(5,8,9),(3,4,0)] [(0,5,7),(0,6,8),(1,5,6),(5,8,10)] 

将不起作用，因为所有3个列表位于同一行上。

这种格式：

[(0,2,3),(0,4,0)] 
[(1,4,5),(5,8,9),(3,4,0)] 
[(0,5,7),(0,6,8),(1,5,6),(5,8,10)] 

将工作，为点的每个列表是一个单独的行。