Index mehrere, nicht nebeneinander liegende Bereiche in numpy

Ich würde gerne wählen Sie mehrere, nicht nebeneinander liegende Bereiche von einem 1d-numpy-array (oder vector).

Angenommen:

>>> idx = np.random.randint(100, size=10)
array([82,  9, 11, 94, 31, 87, 43, 77, 49, 50])

Dies funktioniert natürlich:

>>> idx[0:3]
array([82,  9, 11])

Und diese Werke zu Holen, über einzelne Indizes:

>>> idx[[0,3,4]]
array([82, 94, 31])

Aber was ist, wenn ich wählen Sie die Bereiche 0:3, und 7:?

Ich versucht habe: 

>>> idx[[0:3,7:]]
SyntaxError: invalid syntax

Gibt es eine einfache Möglichkeit, dies zu tun, oder muss ich diese zu generieren, die separat und verketten?

InformationsquelleAutor alexw | 2015-12-09
  1. 22

    Müssen Sie verketten, entweder vor oder nach der Indizierung. np.r_ macht es einfach

    In [116]: idx=np.array([82,  9, 11, 94, 31, 87, 43, 77, 49, 50])
In [117]: np.r_[0:3,7:10]
Out[117]: array([0, 1, 2, 7, 8, 9])
In [118]: idx[np.r_[0:3,7:10]]
Out[118]: array([82,  9, 11, 77, 49, 50])

    np.r_ erweitert die Scheiben und verkettet Sie.

    Können Sie mix-Scheiben und Listen:

    In [120]: np.r_[0:3,7:10,[0,3,4]]
Out[120]: array([0, 1, 2, 7, 8, 9, 0, 3, 4])

    Verkettung vor der Indizierung ist wahrscheinlich schneller als danach, aber für 1d-array wie dieses, ich glaube nicht, dass der Unterschied signifikant ist.

    InformationsquelleAutor hpaulj
Schreibe einen Kommentar