Python numpy fromregex用法及代码示例

本文简要介绍 python 语言中 numpy.fromregex 的用法。

用法:

numpy.fromregex(file, regexp, dtype, encoding=None)

使用正则表达式解析从文本文件构造一个数组。

返回的数组始终是结构化数组，由文件中正则表达式的所有匹配项构成。正则表达式中的组被转换为结构化数组的字段。

参数：

file： 路径或文件

要读取的文件名或文件对象。

regexp： str 或正则表达式

用于解析文件的正则表达式。正则表达式中的组对应于 dtype 中的字段。

dtype： dtype 或 dtype 列表

结构化数组的 Dtype；必须是结构化数据类型。

encoding： str，可选

用于解码输入文件的编码。不适用于输入流。

返回：

output： ndarray

输出数组，包含正则表达式匹配的部分文件内容。输出始终是结构化数组。

抛出：

TypeError

当 dtype 不是结构化数组的有效 dtype 时。

注意：

结构化数组的 Dtype 可以以多种形式指定，但所有形式都至少指定数据类型和字段名称。有关详细信息，请参阅 basics.rec。

例子：

>>> from io import StringIO
>>> text = StringIO("1312 foo\n1534  bar\n444   qux")

>>> regexp = r"(\d+)\s+(...)"  # match [digits, whitespace, anything]
>>> output = np.fromregex(text, regexp,
...                       [('num', np.int64), ('key', 'S3')])
>>> output
array([(1312, b'foo'), (1534, b'bar'), ( 444, b'qux')],
      dtype=[('num', '<i8'), ('key', 'S3')])
>>> output['num']
array([1312, 1534,  444])