Cara mendapatkan indeks nilai maksimum dalam array numpy

Anda dapat menggunakan metode berikut untuk mendapatkan indeks nilai maksimum dalam array NumPy:

Metode 1: Dapatkan indeks nilai maksimum dalam array satu dimensi

 x. argmax ()

Metode 2: Dapatkan indeks nilai maksimum di setiap baris array multidimensi

 x. argmax (axis= 1 )

Metode 3: Dapatkan indeks nilai maksimum di setiap kolom array multidimensi

 x. argmax (axis= 0 )

Contoh berikut menunjukkan cara menggunakan masing-masing metode dalam praktik.

Contoh 1: Dapatkan indeks nilai maksimum dalam array satu dimensi

Kode berikut menunjukkan cara mendapatkan indeks nilai maksimum dalam array NumPy satu dimensi:

 import numpy as np

#create NumPy array of values
x = np. array ([2, 7, 9, 4, 4, 6, 3])

#find index that contains max value
x. argmax ()

2

Fungsi argmax() mengembalikan nilai 2 .

Ini memberitahu kita bahwa nilai pada posisi indeks 2 array berisi nilai maksimum.

Jika kita melihat array aslinya, kita dapat melihat bahwa nilai pada posisi indeks 2 adalah 9 , yang memang merupakan nilai maksimum dalam array tersebut.

Contoh 2: Dapatkan indeks nilai maksimum di setiap baris array multidimensi

Kode berikut menunjukkan cara mendapatkan indeks nilai maksimum di setiap baris array NumPy multidimensi:

 import numpy as np

#create multi-dimentional NumPy array
x = np. array ([[4, 2, 1, 5], [7, 9, 2, 0]])

#view NumPy array
print (x)

[[4 2 1 5]
 [7 9 2 0]]

#find index that contains max value in each row
x. argmax (axis= 1 )

array([3, 1], dtype=int32)

Dari hasilnya kita dapat melihat:

• Nilai maksimum baris pertama terletak pada posisi indeks 3 .
• Nilai maksimum baris kedua terletak pada posisi indeks 1 .

Contoh 3: Dapatkan indeks nilai maksimum di setiap kolom array multidimensi

Kode berikut menunjukkan cara mendapatkan indeks nilai maksimum di setiap kolom array NumPy multidimensi:

 import numpy as np

#create multi-dimentional NumPy array
x = np. array ([[4, 2, 1, 5], [7, 9, 2, 0]])

#view NumPy array
print (x)

[[4 2 1 5]
 [7 9 2 0]]

#find index that contains max value in each column
x. argmax (axis= 0 )

array([1, 1, 1, 0], dtype=int32)

Dari hasilnya kita dapat melihat:

• Nilai maksimum pada kolom pertama terletak pada posisi indeks 1 .
• Nilai maksimum pada kolom kedua terletak pada posisi indeks 1 .
• Nilai maksimum pada kolom ketiga terletak pada posisi indeks 1 .
• Nilai maksimum pada kolom keempat terletak pada posisi indeks 0 .

Terkait: Penjelasan Sederhana tentang Sumbu NumPy

Sumber daya tambahan

Tutorial berikut menjelaskan cara melakukan operasi umum lainnya dengan Python:

Cara mengisi array NumPy dengan nilai
Cara mengganti elemen dalam array NumPy
Cara mendapatkan baris tertentu dari array NumPy