python - 将字符串转换为 numpy 数组

Question

输入:

mystr = "100110"

期望的输出 numpy 数组:

mynumpy == np.array([1, 0, 0, 1, 1, 0])

我试过:

np.fromstring(mystr, dtype=int, sep='')

但问题是我无法将我的字符串拆分为它的每个数字，因此 numpy 将其视为一个数字。知道如何将我的字符串转换为 numpy 数组吗？

Answer 1

list 可以帮助您做到这一点。

import numpy as np

mystr = "100110"
print np.array(list(mystr))
# ['1' '0' '0' '1' '1' '0']

如果你想获取数字而不是字符串:

print np.array(list(mystr), dtype=int)
# [1 0 0 1 1 0]