2013-02-01 55 views
37

我是一個Python的完全新手,但它看起來像給定的字符串能夠(有效)任意長度。即您可以採取string str並繼續添加:str += "some stuff..."。有沒有辦法讓這樣的字符串數組?python numpy任意長度字符串數組

當我嘗試這一點,每個元件僅存儲單個字符

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

在另一方面,我知道可以初始化一定長度的字符串數組,即

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

其可以存儲最多256個字符的10個字符串

回答

68

您可以通過創建一個dtype=object的數組來完成此操作。如果你嘗試長字符串分配到一個正常的numpy的陣列,它截斷字符串:

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

但是當你使用dtype=object,你得到Python對象引用數組。所以,你可以有Python中的所有行爲:

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

事實上,因爲它是對象的數組,你可以指定任何一種蟒蛇對象的數組:

>>> a[2] = {1:2, 3:4} 
>>> a 
array([apples, foobar, {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 
+0

謝謝,你的第一個例子特別有用 - 我從來不會猜到那種行爲!我並不擔心這個對象的速度,所以訪問速度慢應該沒問題。 – DilithiumMatrix

+1

很好的答案。我已經將它與演示的鏈接合併到我正在研究的有關numpy數組創建的python筆記本頁面中。 –

11

您可以使用對象數據類型:

>>> import numpy 
>>> s = numpy.array(['a', 'b', 'dude'], dtype='object') 
>>> s[0] += 'bcdef' 
>>> s 
array([abcdef, b, dude], dtype=object)