python - numpy fromstring 空字符串分隔符

Question

当我打电话时:

np.fromstring('3 3 3 0', sep=' ')

返回

array([ 3.,  3.,  3.,  0.])

由于默认情况下，sep=''，我希望以下调用返回相同的结果:

np.fromstring('3330')

无论怎样提高

ValueError: string size must be a multiple of element size

这是为什么？从 '3330' 获取 array([ 3., 3., 3., 0.]) 的最 pythonic 方式是什么？

Answer 1

你可以使用 np.fromiter :

In [11]: np.fromiter('3300', dtype=np.float64)
Out[11]: array([ 3.,  3.,  0.,  0.])

In [12]: np.fromiter('3300', dtype=np.float64, count=4)  # count=len(..)
Out[12]: array([ 3.,  3.,  0.,  0.])