mindspore.dataset.utils.LineReader

查看源文件
class mindspore.dataset.utils.LineReader(filename)[源代码]

基于行的文件读取器。

通过提前缓存文件基于行的元信息,实现对文件各行的随机访问读取。

参数:
  • filename (str) - 待读取的文件名。

异常:
  • TypeError - filename 无效。

  • RuntimeError - filename 不存在或者不是常规文件。

样例:

>>> from mindspore.dataset import LineReader
>>>
>>> reader = LineReader("/path/to/txt/or/csv/file")
>>> # Read the first line of csv file
>>> reader.readline(1)
>>> # Return the row size in csv file
>>> reader.len()
>>> # Close the handle
>>> reader.close()
close()[源代码]

关闭文件句柄。

len()[源代码]

获取当前文件的总行数。

readline(line)[源代码]

读取指定行的内容。

参数:
  • line (int) - 待读取的行号,起始行号为1。

返回:

str,对应行的内容,不包含换行符。

异常:
  • TypeError - 如果 line 不是int类型。

  • ValueError - 如果 line 取值大于文件总行数。