python - 从两个文件创建 float 的 python 列表

Question

我有两个文件要用于绘图。一个包含 x,y 范围，另一个包含数据。

以下是我使用 CSV 阅读器从一个文件中获取数据的代码。

(我在 range.dat 文件的每一行中都有 x,y\n 格式的数据)

import csv
with open('range.dat', 'r') as f,open('data,dat','r') as d:
         reader = csv.reader(f)
         reader1 = csv.reader(d)
         for row in reader:
             print row
             x, y = map(float, row[0:2])
             print [x,y]

现在的问题是我还想同时从同一行获取来自 data.dat 的数据，并希望将其附加到 [x,y]，这样最终列表将是 [x, y，数据]。有什么方法可以让我在同一个 for 循环中遍历 data.dat 内容。

Answer 1

最好的办法可能是将这两个阅读器zip 在一起。如果你使用的是 Python 2，你应该使用 itertools.izip获得发电机；对于 Python 3，您可以只使用 zip。

for row, data in itertools.izip(reader, reader1):
    x, y = map(float, row[0:2])
    print [x,y, data]

或者，您也可以在其他阅读器上调用 next:

for row in reader:
    data = next(reader1)
    ...