在python 3中使用genfromtxt函数时Numpy抛出错误

时间:2017-12-31 03:16:49

标签: python numpy

我的示例代码将是这样的

import numpy as np
from io import BytesIO

data = "1, 2, 3\n4, 5, 6"
np.genfromtxt(data, delimiter=",")

运行时此代码抛出错误

  

追踪(最近一次通话):     文件"",第1行,in   TypeError:需要类似字节的对象,而不是' str'

1 个答案:

答案 0 :(得分:2)

在阅读之前对字符串进行编码:

data = "1, 2, 3\n4, 5, 6"
np.genfromtxt(BytesIO(data.encode()), delimiter=",")

array([[ 1.,  2.,  3.],
       [ 4.,  5.,  6.]])