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])