Создайте многомерный массив из 1-D списка, numpy - PullRequest
Новая
       37

Создайте многомерный массив из 1-D списка, numpy

0 голосов
/ 15 января 2019

Изо всех сил пытаюсь описать эту проблему словами, но у меня, казалось бы, простая проблема, на которую я не могу найти ответ.

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

import numpy as np

a = np.array([9, 8, 7, 6, 5, 4, 3, 2, 1, 0, -1, -2])

b = [[0, 1], [2, 3, 4], [6, 7, 8, 9, 10]]

result = func(a, b) #some function or operator...

print(result)
[[9, 8], [7, 6, 5], [3, 2, 1, 0, -1]]

Спасибо! :)

EDIT:

Пока хорошие решения, но я бы предпочел сделать это без цикла for, так как мы смотрим на сотни тысяч строк и должны сократить время вычислений. Еще раз спасибо:)

Ответы [ 2 ]

0 голосов
/ 15 января 2019

Просто примените каждый индексный подсписок к a: 

In [483]: a = np.array([9, 8, 7, 6, 5, 4, 3, 2, 1, 0, -1, -2])
     ...: 
     ...: b = [[0, 1], [2, 3, 4], [6, 7, 8, 9, 10]]
     ...: 
     ...: 
In [484]: [a[idx] for idx in b]
Out[484]: [array([9, 8]), array([7, 6, 5]), array([ 3,  2,  1,  0, -1])]

Подсписки различаются по длине, поэтому результат не может быть преобразован в 2d массив - он должен оставаться списком (или если вы настаиваете на массиве dtype объекта 1d).

0 голосов
/ 15 января 2019

Вы можете использовать понимание списка: 

>>> [a[x[0]:x[-1]+1] for x in b]
[array([9, 8]), array([7, 6, 5]), array([ 3,  2,  1,  0, -1])]

РЕДАКТИРОВАТЬ: Ваш вопрос указывает, что вы хотите более быстрый вариант, поэтому вы можете протестировать следующий скрипт, чтобы увидеть, какой из них быстрее для вашей установки Python: 

#!/usr/bin/env python                                                               

import timeit

setup = '''                                                                         
import numpy as np                                                                  

a = np.array([9, 8, 7, 6, 5, 4, 3, 2, 1, 0, -1, -2])                                
b = [[0, 1], [2, 3, 4], [6, 7, 8, 9, 10]]                                           
'''

test1 = '''                                                                         
def test():                                                                         
  return [a[x[0]:x[-1]+1] for x in b]                                               
'''

test2 = '''                                                                         
def test():                                                                         
  return [a[idx] for idx in b]                                                      
'''

print(timeit.timeit(setup = setup,
                    stmt = test1,
                    number = 1000000))

print(timeit.timeit(setup = setup,
                    stmt = test2,
                    number = 1000000))

На моей машине два подхода, данные вам до сих пор, работают примерно одинаково, но answer hpaulj может быть немного быстрее (если Python не кеширует данные за кулисами), что может быть более полезным Вам в производстве. Проверьте это локально и посмотрите, получите ли вы похожий или другой ответ.

...