Первый и последний элемент Python из массива

Я пытаюсь динамически получить первый и последний элемент из массива.

Итак, допустим, массив имеет 6 элементов.

test = [1,23,4,6,7,8]

Если я пытаюсь получить first and last = 1,8, 23,7 и 4,6. Есть ли способ получить элементы в этом порядке? Я рассмотрел пару вопросов Ссылка Link2. Я воспользовался этими ссылками, и я придумал этот прототип.

#!/usr/bin/env python

import numpy

test = [1,23,4,6,7,8]
test1 = numpy.array([1,23,4,6,7,8])
len_test = len(test)
first_list = [0,1,2]
len_first = len(first_list)
second_list = [-1,-2,-3]
len_second = len(second_list)

for a in range(len_first):
        print numpy.array(test)[[first_list[a] , second_list[a]]]
        print test1[[first_list[a], second_list[a]]]

Но этот прототип не будет масштабироваться, если у вас есть более 6 элементов. Итак, мне было интересно, есть ли способ динамически получить пару элементов.

Спасибо!

Ответ 1

Как насчет:

In [10]: arr = numpy.array([1,23,4,6,7,8])

In [11]: [(arr[i], arr[-i-1]) for i in range(len(arr) // 2)]
Out[11]: [(1, 8), (23, 7), (4, 6)]

В зависимости от размера arr запись всей вещи в NumPy может быть более эффективной:

In [41]: arr = numpy.array([1,23,4,6,7,8]*100)

In [42]: %timeit [(arr[i], arr[-i-1]) for i in range(len(arr) // 2)]
10000 loops, best of 3: 167 us per loop

In [43]: %timeit numpy.vstack((arr, arr[::-1]))[:,:len(arr)//2]
100000 loops, best of 3: 16.4 us per loop

Ответ 2

Я закончил здесь, потому что я googled для "python первый и последний элемент массива" и нашел все остальное, кроме этого. Итак, вот ответ на титульный вопрос:

a = [1,2,3]
a[0] # first element (returns 1)
a[-1] # last element (returns 3)

Ответ 3

>>> test = [1,23,4,6,7,8]
>>> from itertools import izip_longest
>>> for e in izip_longest(test, reversed(test)):
    print e


(1, 8)
(23, 7)
(4, 6)
(6, 4)
(7, 23)
(8, 1)

Другой вариант

>>> test = [1,23,4,6,7,8]
>>> start, end = iter(test), reversed(test)
>>> try:
    while True:
        print map(next, [start, end])
except StopIteration:
    pass

[1, 8]
[23, 7]
[4, 6]
[6, 4]
[7, 23]
[8, 1]

Ответ 4

Использование индексирования Numpy:

>>> test
array([ 1, 23,  4,  6,  7,  8])

>>> test[::-1]  # test, reversed
array([ 8,  7,  6,  4, 23,  1])

>>> numpy.vstack([test, test[::-1]])  # stack test and its reverse
array([[ 1, 23,  4,  6,  7,  8],
       [ 8,  7,  6,  4, 23,  1]])

>>> # transpose, then take the first half;
>>> # +1 to cater to odd-length arrays
>>> numpy.vstack([test, test[::-1]]).T[:(len(test) + 1) // 2]
array([[ 1,  8],
       [23,  7],
       [ 4,  6]])

vstack копирует массив, но все остальные операции представляют собой трюки указателя времени (включая изменение) и, следовательно, очень быстрые.

Ответ 5

Как насчет этого?

>>> import numpy
>>> test1 = numpy.array([1,23,4,6,7,8])
>>> forward = iter(test1)
>>> backward = reversed(test1)
>>> for a in range((len(test1)+1)//2):
...     print forward.next(), backward.next()
... 
1 8
23 7
4 6

(len(test1)+1)//2 гарантирует, что средний элемент массивов нечетной длины также возвращается:

>>> test1 = numpy.array([1,23,4,9,6,7,8]) # additional element '9' in the middle
>>> forward = iter(test1)                                                      
>>> backward = reversed(test1)
>>> for a in range((len(test1)+1)//2):
...     print forward.next(), backward.next()
1 8
23 7
4 6
9 9

Использование только len(test1)//2 приведет к сбросу средних элементов массивов нечетной длины.

Ответ 6

Это делает. Обратите внимание, что при нечетном числе элементов один из них не будет включен.

test = [1, 23, 4, 6, 7, 8, 5]    
for i in range(len(test)/2):
    print (test[i], test[-1-i])

Вывод:

(1, 5)
(23, 8)
(4, 7)

Ответ 7

Вы можете просто использовать метод take и индекс элемента (последний индекс может быть -1).

arr = np.array([1,2,3])

last = arr.take(-1)
# 3

Ответ 8

Предполагая, что в списке есть четное число элементов, вы можете сделать:

test = [1,23,4,6,7,8]
test_rest = reversed(test[:len(test)/2])

for n in len(test_rest):
    print [test[n], test_test[n]]