如何创建任意长度字符串的numpy数组？

发布时间：2020-12-20 10:34:51 所属栏目：Python 来源：网络整理

导读：我是 Python的新手,但似乎给定的字符串能够(有效地)任意长度.即你可以采用字符串str并继续添加它：str =“some stuff ……”.有没有办法制作这样的字符串数组？当我尝试这个时,每个元素只存储一个字符 strArr = numpy.empty(10,dtype='string')for i in ran

我是 Python的新手,但似乎给定的字符串能够(有效地)任意长度.即你可以采用字符串str并继续添加它：str =“some stuff ……”.有没有办法制作这样的字符串数组？

当我尝试这个时,每个元素只存储一个字符

strArr = numpy.empty(10,dtype='string')
for i in range(0,10)
    strArr[i] = "test"

另一方面,我知道我可以初始化一定长度字符串的数组,即

strArr = numpy.empty(10,dtype='s256')

它可以存储10个最多256个字符的字符串.

解决方法

您可以通过创建dtype = object数组来完成此操作.如果您尝试将长字符串分配给普通的numpy数组,它会截断字符串：

>>> a = numpy.array(['apples','foobar','cowboy'])
>>> a[2] = 'bananas'
>>> a
array(['apples','banana'],dtype='|S6')

但是当你使用dtype = object时,你会得到一个python对象引用数组.所以你可以拥有python字符串的所有行为：

>>> a = numpy.array(['apples','cowboy'],dtype=object)
>>> a
array([apples,foobar,cowboy],dtype=object)
>>> a[2] = 'bananas'
>>> a
array([apples,bananas],dtype=object)

实际上,因为它是一个对象数组,所以你可以将任何类型的python对象分配给数组：

>>> a[2] = {1:2,3:4}
>>> a
array([apples,{1: 2,3: 4}],dtype=object)

然而,这解决了使用numpy的许多好处,这是如此之快,因为它适用于大量连续的原始内存块.使用python对象会增加很多开销.一个简单的例子：

>>> a = numpy.array(['abba' for _ in range(10000)])
>>> b = numpy.array(['abba' for _ in range(10000)],dtype=object)
>>> %timeit a.copy()
100000 loops,best of 3: 2.51 us per loop
>>> %timeit b.copy()
10000 loops,best of 3: 48.4 us per loop

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!