您不需要这些行延续反斜杠-现代Python足够聪明,可以发现列表定义不完整,因此它会自动为您执行行延续。在
另外,不需要导入数学模块来计算平方根,可以使用内置的幂运算符:** 0.5。在
要获得所需的输出,需要将nList中的跟踪列表一次提供给生成器,将生成的数据捕获到列表中,然后保存在all_lengths_list中。这可以在普通的for循环中完成,但使用列表理解更紧凑,如下面的代码所示。为了显示all_lengths_list的内容,我使用pprint()来自{a1}模块。虽然不是很漂亮,但总比把它们放在一条线上要好。:)#!/usr/bin/env python
from pprint import pprint
nList = [
[[0, 0, 0], [3, 2, 1]],
[[]],
[[1, 1, 12]],
[[0, 0, 0], [30000, 40, 3010], [32000, 40500, 7520], [0, 0, 10520]],
[[15340, 0, 0], [104300, 0, 3630], [434000, 4434000, 63460]]
]
def calculate_length(x1, y1, x2, y2):
return ((x1-x2)**2 + (y1 - y2)**2) ** 0.5
def calculate_time (t1,t2):
return abs(t1-t2)
def lengths(trace):
previous_x, previous_y = 0, 0
for index, point in enumerate(trace):
if point:
x, y, t = point
if index > 0:
yield calculate_length(x, y, previous_x, previous_y)
previous_x, previous_y = x, y
all_lengths_list = [list(lengths(trace)) for trace in nList]
pprint(all_lengths_list, indent=4)
输出
^{pr2}$