Какой самый простой способ создать массив значений индекса в Python? ⇐ Python

[Anonymous]

При наличии некоторых данных в массиве numpy, каков наиболее эффективный способ создания аналогичного массива numpy, содержащего значения индекса?
Я рассмотрел несколько возможностей, подробности которых приведены ниже. Мне интересно, есть ли более простой метод.
1. numpy.linspace

Код: Выделить всё

numpy.linspace(0, len(data_array), len(data_array), endpoint=False)

Это можно немного изменить, чтобы возвращать целочисленные значения, а не значения с плавающей запятой, если это соответствует проблемной области.

Код: Выделить всё

numpy.linspace(0, len(data_array), len(data_array), endpoint=False, dtype=Int)

2. перечислить

Код: Выделить всё

enumerate

может использоваться в различных формах. Я предпочитаю форму с использованием карты.

Код: Выделить всё

numpy.array([index for (index, _) in enumerate(data_array)])

Код: Выделить всё

numpy.array(list(map(lambda pair: pair[0], enumerate(data_array)))

3. numpy.indices
Этот вариант выглядит некрасиво из-за дополнительной запятой, необходимой для создания кортежа.

Код: Выделить всё

numpy.indices((len(price_data),))[0]

Каждый из вышеперечисленных методов возвращает правильные значения. Однако каждый из них несколько громоздок для такой простой задачи.

Подробнее здесь: https://stackoverflow.com/questions/792 ... -in-python