python - numpy.loadtxt 跳过多行

Question

我相信这个线程的标题解释了我在寻找什么。我很想知道跳过多行的语法是什么；我似乎无法在任何地方找到此类信息。

Answer 1

使用帮助(np.loadtxt)。您会发现 skiprows 参数将允许您跳过前 N 行:

In [1]: import numpy as np

In [2]: help(np.loadtxt)
Help on function loadtxt in module numpy.lib.npyio:

loadtxt(fname, dtype=<type 'float'>, comments='#', delimiter=None, converters=None, skiprows=0, usecols=None, unpack=False, ndmin=0)
    ...
    skiprows : int, optional
        Skip the first `skiprows` lines; default: 0.

因此，要跳过 N 行，您会说

np.loadtxt(fname, skiprows=N)

---

如果您需要过滤除first N 行以外的行，请使用np.genfromtxt，它可以采用迭代器生成字符串作为它的第一个参数:

with open(filename, 'r') as f:
    lines = (line for line in f if predicate(line))
    arr = np.genfromtxt(lines)

---

要跳过中间的一系列行，例如第 47--50 行，您可以像这样使用 itertools:

import itertools as IT

with open(filename, 'r') as f:
    lines = IT.chain(IT.islice(f, 46), IT.islice(f, 4, None))
    arr = np.genfromtxt(lines)