numpy字符串的字符串

时间:2012-06-11 17:02:01

标签: python arrays string numpy char

我有一个2D numpy char数组(来自NetCDF4文件),它实际上代表了一个字符串列表。我想将其转换为字符串列表。

我知道我可以使用join()将chars连接成一个字符串,但我只能找到一种方法来一次执行一个字符串:

data = np.array([['a','b'],['c','d']])
for row in data[:]:
    print ''.join(row)

但它很慢。如何在单个命令中返回字符串数组?感谢

3 个答案:

答案 0 :(得分:12)

列表理解是最“pythonic”的方式。

最“numpythonic”的方式是:

>>> data = np.array([['a','b'],['c','d']])
# a 2D view
>>> data.view('S2')
array([['ab'],
       ['cd']], 
      dtype='|S2')
# or maybe a 1D view ...fastest solution:
>>> data.view('S2').ravel()
array(['ab', 'cd'], 
      dtype='|S2')

没有循环,没有列表理解,甚至没有副本。缓冲区只是保持不变,具有不同的“视图”,因此这是最快的解决方案。

答案 1 :(得分:3)

尝试列表理解:

>> s = [''.join(row) for row in data]
>> s
['ab', 'cd']

这只是你的for循环重写。

答案 2 :(得分:2)

[row.tostring() for row in data]