Numpy 对象和字符串相互转换

在实际工作中有个需求是需要将 Numpy 的二进制对象转换为字符串,经过某种处理之后,再将字符串还原为 Numpy 对象,这就需要用到 Numpy 自带的 tostringfromstring 方法。在此记录下其使用方法。

1. tostring 方法

numpy 对象转换为字符串:

In [1]: import numpy as np

In [2]: a = np.array([[1,2], [3,4]])

In [3]: a
Out[3]: 
array([[1, 2],
       [3, 4]])

In [4]: b = a.tostring()

In [5]: b
Out[5]: b'\x01\x00\x00\x00\x02\x00\x00\x00\x03\x00\x00\x00\x04\x00\x00\x00'

In [6]: 

2. fromstring 方法

fromstring 或者 frombuffer 可以将字符串对象转换为对应的 numpy 对象。

In [7]: a.dtype
Out[7]: dtype('int32')

In [8]: c = np.fromstring(b, dtype=np.int32)
g:\python\lib\site-packages\ipykernel_launcher.py:1: DeprecationWarning: The binary mode of fromstring is deprecated, as it behaves surprisingly on unicode inputs. Use frombuffer instead
  """Entry point for launching an IPython kernel."""

In [9]: c = np.frombuffer(b, dtype=np.int32)

In [10]: c
Out[10]: array([1, 2, 3, 4])
In [11]: c.shape=(2,2)

In [12]: c
Out[12]: 
array([[1, 2],
       [3, 4]])

In [13]: 

注意:Python3 中更推荐使用 frombuffer 来将字符串转换为二进制,并且要设置转换的 dtype 类型,否则默认按照一维进行转换

3. 文本文件处理

对于文本文件,推荐使用

  • loadtxt
  • genfromtxt
  • savetxt

4. 二进制文件处理

对于二进制文本文件,推荐使用

  • save
  • load
  • savez
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章