如何控制 genfromtxt 读取指定的行？

Question

genfromtxt 可以跳过页眉和页脚行并指定要使用的列。 但是如何控制要读取的行数？

有时一个 txt 文件可能包含几个不同形状的块。 例如，

a=StringIO('''
1,2,3
1,2,3
2,3
2,3
''')
genfromtxt(a,delimiter=',',skip_header=1)

这会引发错误，

ValueError: Some errors were detected !
    Line #4 (got 2 columns instead of 3)
    Line #5 (got 2 columns instead of 3)

当然，我可以这样做：

a=StringIO('''
1,2,3
1,2,3
2,3
2,3
''')
genfromtxt(a,delimiter=',',skip_header=1,skip_footer=2)

这很丑，因为我必须计算块下的行数。

但是我希望像

genfromtxt(a,delimiter=',',skip_header=1,nrows=2)

这样会更清楚。

有人对此有好的想法吗？还是使用其他功能？

---

2015 年 10 月更新

这个问题已经在Numpy的新版本中解决。

genfromtxt 现在有一个名为max_rows 的新关键字，可以控制要读取的行数，参见here。

Answer 1

您可以使用invalid_raise = False 跳过读取缺少某些数据的行。 例如

b = np.genfromtxt(a, delimiter=',', invalid_raise=False)

这会给你一个警告，但不会引发异常。