Bir liste veya tuple'dan açıkça öğe seçin

120

Aşağıdaki Python listesine sahibim (ayrıca bir başlık da olabilir):

myList = ['foo', 'bar', 'baz', 'quux']

söyleyebilirim

>>> myList[0:3]
['foo', 'bar', 'baz']
>>> myList[::2]
['foo', 'baz']
>>> myList[1::2]
['bar', 'quux']

Endeksleri belirli kalıpları olmayan öğeleri açıkça nasıl seçerim? Örneğin, seçmek istiyorum [0,2,3]. Ya da 1000 öğelik çok büyük bir listeden seçmek istiyorum [87, 342, 217, 998, 500]. Bunu yapan bir Python sözdizimi var mı? Şuna benzeyen bir şey:

>>> myBigList[87, 342, 217, 998, 500]

— malzeme
kaynak

1

Bu bir kopya gibi görünüyor. Diğer sorunun daha fazla oyu var ama bu zamanlamayla daha iyi bir cevabı var gibi görünüyor.

— AnnanFay

150

list( myBigList[i] for i in [87, 342, 217, 998, 500] )

Cevapları python 2.5.2 ile karşılaştırdım:

19.7 usec: [ myBigList[i] for i in [87, 342, 217, 998, 500] ]
20.6 usec: map(myBigList.__getitem__, (87, 342, 217, 998, 500))
22.7 usec: itemgetter(87, 342, 217, 998, 500)(myBigList)
24.6 usec: list( myBigList[i] for i in [87, 342, 217, 998, 500] )

Python 3'te 1.'in 4. ile aynı olacak şekilde değiştirildiğini unutmayın.

Diğer bir seçenek, numpy.arraybir liste veya a yoluyla indekslemeye izin veren bir ile başlamaktır numpy.array:

>>> import numpy
>>> myBigList = numpy.array(range(1000))
>>> myBigList[(87, 342, 217, 998, 500)]
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
IndexError: invalid index
>>> myBigList[[87, 342, 217, 998, 500]]
array([ 87, 342, 217, 998, 500])
>>> myBigList[numpy.array([87, 342, 217, 998, 500])]
array([ 87, 342, 217, 998, 500])

tupleBu dilimler gibi aynı şekilde çalışmaz.

— Dan D.
kaynak

2

Tercihen bir liste kompozisyonu olarak [myBigList[i] for i in [87, 342, 217, 998, 500]], ancak bu yaklaşımı en çok seviyorum.

— zeekay

@MedhatHelmy Bu zaten cevapta. from operator import itemgetterBaşlatma kısmında kullanılan üçüncü seçenek python -mtimeit.

— Dan D.

Merak ediyorum, sadece bir dil tasarımı perspektifinden bakarsak, normal bir python myBigList[(87, 342, 217, 998, 500)]iken neden çalışmıyor ? Bunu denediğimde anladım . Bu, anlamayı yazmaktan çok daha kolay olurdu - ilgili bir dil tasarımı / uygulaması sorunu var mı? myBigListlistTypeError: list indices must be integers or slices, not tuple

— sparc_spread

@sparc_spread, bunun nedeni listsPython'da yalnızca tam sayıları veya dilimleri kabul etmesidir . Bir tamsayı geçirmek, mevcut bir listeden yalnızca bir öğenin alınmasını sağlar. Bir dilim iletmek, bir parçasının geri alınmasını sağlar, ancak bir demet iletmek, bir veri türünü ( tuple) bağımsız değişken olarak başka bir veri türüne ( ) iletmek gibidir ve listbu da sözdizimsel olarak yanlıştır.

— amanb

48

Peki buna ne dersin:

from operator import itemgetter
itemgetter(0,2,3)(myList)
('foo', 'baz', 'quux')

— Marcin
kaynak

2

Şimdiye kadarki en seksi bu. Bu operatormodülü sevdim!

— jathanism

10

Yerleşik değildir, ancak isterseniz tupleları "dizinler" olarak alan bir liste alt sınıfı oluşturabilirsiniz:

class MyList(list):

    def __getitem__(self, index):
        if isinstance(index, tuple):
            return [self[i] for i in index]
        return super(MyList, self).__getitem__(index)


seq = MyList("foo bar baaz quux mumble".split())
print seq[0]
print seq[2,4]
print seq[1::2]

baskı

foo
['baaz', 'mumble']
['bar', 'quux']

— Matt Anderson
kaynak

2

(+1) Güzel çözüm! Bu uzantı ile Python'da dizileri işlemek çok R veya Matlab gibi görünmeye başlar.

— Assad Ebrahim

7

Belki bir liste anlayışı sırayla:

L = ['a', 'b', 'c', 'd', 'e', 'f']
print [ L[index] for index in [1,3,5] ]

üretir:

['b', 'd', 'f']

Aradığın şey bu mu?

— Dan Witkowski
kaynak

6

>>> map(myList.__getitem__, (2,2,1,3))
('baz', 'baz', 'bar', 'quux')

Ayrıca, yapabilmek istiyorsanız List, demetleri bağımsız değişken olarak destekleyen kendi sınıfınızı da oluşturabilirsiniz .__getitem__myList[(2,2,1,3)]

— ninjagecko
kaynak

Bu işe yarasa da, sihirli değişkenleri doğrudan çağırmak genellikle iyi bir fikir değildir. Bir liste anlama veya benzeri bir yardımcı modül kullanmakta daha iyi olursunuz operator.

— jathanism

@jathanism: Saygıyla katılmıyorum. Gerçi, ileriye dönük uyumluluk konusunda endişeleriniz varsa (genel / özelden farklı olarak) nereden geldiğinizi kesinlikle görebiliyorum.

— ninjagecko

Ben oradan geliyorum. :) Bunu takiben, kullanmanın daha iyi olmasının nedeni aynılen(myList) üzerinde myList.__len__().

— jathanism

Yaratıcı bir çözüm Sihirli değişkeni çağırmanın kötü bir fikir olduğunu düşünmüyorum. programcı, programlama koşullarına göre tercih ettiği yolu seçer.

— Jacob CUI

2

Sadece şunu belirtmek istiyorum, itemgetter'ın sözdizimi bile gerçekten düzgün görünüyor, ancak büyük listede icra edildiğinde biraz yavaş.

import timeit
from operator import itemgetter
start=timeit.default_timer()
for i in range(1000000):
    itemgetter(0,2,3)(myList)
print ("Itemgetter took ", (timeit.default_timer()-start))

Itemgetter 1.065209062149279 aldı

start=timeit.default_timer()
for i in range(1000000):
    myList[0],myList[2],myList[3]
print ("Multiple slice took ", (timeit.default_timer()-start))

Birden çok dilim 0.6225321444745759 aldı

— Wendao Liu
kaynak

İlk snippet, lütfen ekleyin myList = np.array(range(1000000))aksi takdirde hata alırsınız.

— Cloud Cho

1

Başka bir olası çözüm:

sek=[]
L=[1,2,3,4,5,6,7,8,9,0]
for i in [2, 4, 7, 0, 3]:
   a=[L[i]]
   sek=sek+a
print (sek)

— fdante
kaynak

0

gibi bir boolean numpy dizisine sahip olduğunuzda olduğu gibi mask

[mylist[i] for i in np.arange(len(mask), dtype=int)[mask]]

Herhangi bir dizi veya np.array için çalışan bir lambda:

subseq = lambda myseq, mask : [myseq[i] for i in np.arange(len(mask), dtype=int)[mask]]

newseq = subseq(myseq, mask)

— theo olsthoorn
kaynak