Эффективный способ «отменить» индексы от np.tril_indices

Question

Эффективный способ «отменить» индексы от np.tril_indices

1

np.tril_indices(3) возвращает нижние треугольные индексы квадратной матрицы размером 3:

(array([0, 1, 1, 2, 2, 2], dtype=int64),
array([0, 0, 1, 0, 1, 2], dtype=int64))

Каков наиболее эффективный способ вместо этого получить:

(array([0, 1, 1, 2, 2, 2], dtype=int64),
array([0, 1, 0, 2, 1, 0], dtype=int64))

Благодарю.

Jack 17 июнь 2018, в 16:58

Источник

Теги:

python

arrays

numpy

indexing

2 ответа

4

Один из способов

>>> np.subtract.accumulate(np.tril_indices(3), 0)
array([[0, 1, 1, 2, 2, 2],
       [0, 1, 0, 2, 1, 0]])

Paul Panzer 17 июнь 2018, в 12:16

Ещё вопросы

Divakar · Accepted Answer · 2018-06-17T13-40-00.000Z

Вот один с cumsum и maximum.accumulate -

def reverse_tril_indices(n):
    r = np.arange(n)
    rr = np.arange(n,0,-1)
    N = n*(n+1)//2

    shifts1 = rr[:-1].cumsum()
    shifts2 = r.cumsum()

    id_arr = np.ones(N,dtype=int)    
    id_arr[shifts1] = -rr[1:]
    id_arr[0] = 0
    id1 = id_arr.cumsum()[::-1]

    id2_arr = np.zeros(N,dtype=int) # use dtype=np.uint16 for further boost
    id2_arr[shifts2] = r
    id2 = np.maximum.accumulate(id2_arr)
    return id2, id1

Пример прогона -

In [75]: reverse_tril_indices(3)
Out[75]: (array([0, 1, 1, 2, 2, 2]), array([0, 1, 0, 2, 1, 0]))

In [77]: reverse_tril_indices(5)
Out[77]: 
(array([0, 1, 1, 2, 2, 2, 3, 3, 3, 3, 4, 4, 4, 4, 4]),
 array([0, 1, 0, 2, 1, 0, 3, 2, 1, 0, 4, 3, 2, 1, 0]))

Сроки на n=5k -

# @Paul Panzer soln
In [83]: %timeit np.subtract.accumulate(np.tril_indices(5000), 0)
1 loop, best of 3: 278 ms per loop

In [84]: %timeit reverse_tril_indices(5000)
10 loops, best of 3: 83.4 ms per loop

Ух ты. Гениальное спасибо! Так быстро.