使用列表理解读取文件中的所有行

Question

我有一个简单的问题，如果我在 stackoverflow 中发帖，我很抱歉。我是python的新手，我不记得我如何在列表压缩中读取x，y，z

我的文件是一个 x,y,z 文件，其中每一行都是一个点：

x1,y1,z1
x2,y2,z2
x3,y3,z3
........

inFile = "Myfile.las"

with lasfile.File(inFile, None, 'r') as f:
     # missing part
     points =[]

我希望保存一个只有 x 和 y 的对象

在此先感谢您对这个简单的问题表示歉意

Answer 1

您想要一个 x 和 y 坐标列表，这很简单：

with lasfile.File(inFile, None, 'r') as f:
     # missing part
     points = [line.split(',')[:2] for line in lasfile]

如果这些坐标是整数，您可以通过快速调用map() 将它们转换为python int（来自str）：

points = [map(int, line.split(',')[:2]) for line in lasfile]

在 python 3 中，map 是一个生成器，最好使用嵌套列表推导：

points = [[int(i) for i in line.split(',')[:2]] for line in lasfile]

这将产生一个列表列表：

[[x1, y1], [x2, y2], ...]