python - 使用 numpy 的 genfromtxt 读取每个第 n 行的最快方法

我用 numpy 的 genfromtxt 读取我的数据:

import numpy as np
measurement = np.genfromtxt('measurementProfile2.txt', delimiter=None, dtype=None, skip_header=4, skip_footer=2, usecols=(3,0,2))
rows, columns = np.shape(measurement)
x=np.zeros((rows, 1), dtype=measurement.dtype)
x[:]=394
measurement = np.hstack((measurement, x))
np.savetxt('measurementProfileFormatted.txt',measurement)

这工作正常。但我只想要最终输出文件中的 5-th 、 6-th (所以 n-th )行。
根据 numpy.genfromtxt.html 没有参数可以做到这一点。我不想迭代数组。有没有推荐的方法来处理这个问题？

最佳答案

为了避免读取整个数组，您可以将 np.genfromtxt 与 itertools.islice 结合使用以跳过行。这比读取整个数组然后切片(至少对于我尝试过的较小数组)略快。

例如，这是 file.txt 的内容:

12
34
22
17
41
28
62
71

然后例如:

>>> import itertools
>>> with open('file.txt') as f_in:
        x = np.genfromtxt(itertools.islice(f_in, 0, None, 3), dtype=int)

返回一个数组 x ，其中包含上述文件的 0 、 3 和 6 索引元素:

array([12, 17, 62])

关于python - 使用 numpy 的 genfromtxt 读取每个第 n 行的最快方法，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/27961782/