python - Numpy savetxt 到一个字符串

Question

我想将 numpy.savetxt 的结果加载到一个字符串中。本质上是以下没有中间文件的代码:

import numpy as np

def savetxts(arr):
    np.savetxt('tmp', arr)
    with open('tmp', 'rb') as f:
        return f.read()

Answer 1

对于 Python 3.x，您可以使用 io 模块:

>>> import io
>>> s = io.BytesIO()
>>> np.savetxt(s, (1, 2, 3), '%.4f')
>>> s.getvalue()
b'1.0000\n2.0000\n3.0000\n'

>>> s.getvalue().decode()
'1.0000\n2.0000\n3.0000\n'

注意:我无法让 io.StringIO() 工作。有什么想法吗？