python输入数字成数组_python – 将numpy字符串字段数组转换为数字格式

最新版本的numpy 1.16添加了structured_to_unstructured,它解决了这个问题:

from numpy.lib.recfunctions import structured_to_unstructured

y = structured_to_unstructured(x) # 2d array of 'S2'

z = y.astype(np.int8)

在以前的numpy版本中,您可以将x.data和np.frombuffer组合在一起,从内存中的相同数据创建另一个数组,而不必使用步幅.但它并没有带来性能提升,因为计算是由从S2到int8的转换驱动的.

n = 1000

def f1(x):

y = np.lib.stride_tricks.as_strided(x.view(dtype='S2'),

shape=(n, 3),

strides=(6, 2))

return y.astype(np.int8)

def f2(x):

y = np.frombuffer(x.data, dtype='S2').reshape((n, 3))

return y.astype(np.int8)

x = np.array([(i%3-1, (i+1)%3-1, (i+2)%3-1)

for i in xrange(n)],

dtype='S2,S2,S2')

z1 = f1(x)

z2 = f2(x)

assert (z1==z2).all()

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值