Get Array Zeilen- und Spaltennummer nach Werten in Python

Question

Zum Beispiel habe ich eine 5*5 np.array wie dieses:

a=[[1,2,3,4,5], 
    [6,7,8,9,10], 
    [11,12,13,14,15], 
    [16,17,18,19,20], 
    [21,22,23,24,25]] 

wenn ich den Bereich der Zeile und Spalte, in number<=15 erhalten möchten, wie kann ich das tun?

Im Gegenteil, wenn ich weiß, den Bereich der Zeile und Spalte, wie i in xrange(1,4) und j in xrange(1,4), wie kann ich die Nummer wie bekommen:

[[7,8,9], 
[12,13,14], 
[17,18,19]] 

Answer 1

Um den Bereich auf der Grundlage einer Bedingung zu erhalten Sie können entweder den Zustand direkt anwenden oder np.where verwenden:

>>> a 
array([[ 1, 2, 3, 4, 5], 
     [ 6, 7, 8, 9, 10], 
     [11, 12, 13, 14, 15], 
     [16, 17, 18, 19, 20], 
     [21, 22, 23, 24, 25]]) 
>>> a < 15 
array([[ True, True, True, True, True], 
     [ True, True, True, True, True], 
     [ True, True, True, True, False], 
     [False, False, False, False, False], 
     [False, False, False, False, False]], dtype=bool) 
>>> np.where(a < 15) 
(array([0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 2, 2, 2, 2]), 
array([0, 1, 2, 3, 4, 0, 1, 2, 3, 4, 0, 1, 2, 3])) 

im letzteren Fall ist der Rückgabewert ein Tupel der Anpassungsindizes ist.

die entgegengesetzte Operation zu erreichen, können Sie einfach Ihre Array-Slice:

>>> ar[1:4, 1:4] 
array([[ 7, 8, 9], 
     [12, 13, 14], 
     [17, 18, 19]])