Токенизация целых чисел в строке
У меня есть текстовый файл, содержащий координаты в виде:
[-1.38795678, 54.90352965]
[-3.2115, 55.95530556]
[0.00315428, 51.50285246]
Я хочу иметь возможность перебирать каждую координату, чтобы проверить, в каком полигоне она находится (графства Великобритании в шейп-файле), однако я не уверен, как маркировать числа так, чтобы у меня был код вдоль строк...
for line in coordinates:
for poly in polygons:
if points in polygons:
print(polygons)
break
if points not in polygons:
continue
На данный момент они являются строками, но я хочу, чтобы каждая строка состояла из двух точек, чтобы я мог попытаться найти их в многоугольнике.
2 ответа
Решение
Вы можете превратить строку в кортеж, используя literal_eval
,
>>> from ast import literal_eval
>>> s = "[-1.38795678, 54.90352965], [-3.2115, 55.95530556], [0.00315428, 51.50285246]"
>>> seq = literal_eval(s)
>>> print seq[0][1]
54.90352965
Изменить: если координаты находятся на отдельных строках без запятых,
from ast import literal_eval
s = """[-1.38795678, 54.90352965]
[-3.2115, 55.95530556]
[0.00315428, 51.50285246]"""
seq = [literal_eval(line) for line in s.split("\n")]
#or
seq = literal_eval(s.replace("\n", ","))
print seq[0][1]
Вы также можете использовать регулярное выражение, которое будет значительно быстрее, чем ast:
import re
with open("in.txt") as f:
r = re.compile("[-]?\d+\.\d+")
data = [list(map(float, r.findall(line))) for line in f]
Некоторые сроки:
In [14]: %%timeit
with open("test.txt") as f:
data = [literal_eval(line) for line in f]
....:
100 loops, best of 3: 2.01 ms per loop
In [15]: %%timeit
with open("test.txt") as f:
r = re.compile("[-]\d+\.\d+")
data = [list(map(float, r.findall(line))) for line in f]
....:
1000 loops, best of 3: 403 µs per loop
with open("test.txt") as f:
r = re.compile("[-]?\d+\.\d+")
data = [list(map(float, r.findall(line))) for line in f]
....:
In [38]: with open("test.txt") as f:
data2 = [literal_eval(line) for line in f]
....:
In [39]: data == data2
Out[39]: True
Просто разделение и разделение будет быстрее снова:
In [40]: %%timeit
....: with open("test.txt") as f:
....: data = [list(map(float, line.strip("[]\n").split(","))) for line in f]
....:
1000 loops, best of 3: 249 µs per loop