Як створити масив numpy довільних рядків довжини?


78

Я повністю новачок у Python, але, схоже, даний рядок може бути (фактично) довільною довжиною. тобто ви можете взяти string 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 символів.

Відповіді:


126

Ви можете зробити це, створивши масив 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)

Дійсно, оскільки це масив об’єктів, ви можете призначити масиву будь- який об’єкт python:

>>> 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

2
Дякую, ваш перший приклад особливо корисний - я ніколи не здогадався б про таку поведінку! Мене не турбує швидкість цього об’єкта, тому повільний доступ повинен бути нормальним.
DilithiumMatrix

1
Приємна відповідь. Я включив посилання на нього з демо-версією на сторінку блокнота python, над якою працюю над створенням масиву numpy.
Джон Локвуд,

@senderle, якщо масив має objectdtype, тоді np.fromstring(arr.tostring())не вдасться numpy Cannot create an object array from a string. Будь-які ідеї вирішити це?
youkaichao

@ 游 凯 超 хм. Це важко. Це не повний сюрприз, оскільки numpy насправді не призначений для роботи з об'єктами python. Це швидше ярлик, ніж правильне використання numpy. Тож немає жодної реальної причини для того, щоб підтримувати подібні кутові справи. Моїм підходом, мабуть, було б отримати максимальну довжину рядка та використовувати стандартний масив char фіксованої ширини.
senderle

@ 游 凯 超, якщо ваша мета - використовувати рядки як мітки рядків або заголовків стовпців, вам слід також переглянути структуровані масиви .
senderle

17

Ви можете використовувати тип даних об'єкта:

>>> import numpy
>>> s = numpy.array(['a', 'b', 'dude'], dtype='object')
>>> s[0] += 'bcdef'
>>> s
array([abcdef, b, dude], dtype=object)
Використовуючи наш веб-сайт, ви визнаєте, що прочитали та зрозуміли наші Політику щодо файлів cookie та Політику конфіденційності.
Licensed under cc by-sa 3.0 with attribution required.