深入淺出np.amin函數

一、概述

NumPy是Python數據分析中最常用的庫之一。NumPy中的amin（）函數可以用於從給定數組中返回最小值，即它返回數組中的最小值。此函數還可以沿指定的軸比較兩個數組，並返回較小值的數組元素。

import numpy as np

# 一維數組
arr = np.array([1, 2, 3, 4, 5])
print('一維數組arr的最小值為：', np.amin(arr))

# 二維數組
arr2d = np.array([[1, 2], [3, 4]])
print('二維數組arr2d的最小值為：', np.amin(arr2d, axis=0))

上述代碼中，我們首先導入了NumPy庫，並定義了一個一維數組arr和一個二維數組arr2d，然後將它們作為參數傳遞給amin()函數。

二、返回值類型

np.amin()函數的返回值取決於傳遞給它的參數類型。如果向amin()函數傳遞單個一維或二維數組，則返回其最小值。否則，它將比較兩個數組，並返回較小值的數組元素。

如果參數是單個一維或二維數組，則np.amin()函數返回一個標量值。否則，它將返回一個形狀相同的數組。

# 返回標量值
arr = np.array([1, 2, 3, 4, 5])
print('一維數組arr的最小值為：', np.amin(arr))

# 返回數組
arr2d = np.array([[1, 2], [3, 4]])
arr2d_min = np.amin(arr2d, axis=0)
print('二維數組arr2d每列的最小值為：', arr2d_min) 

三、軸參數

amin()函數可以沿指定的軸比較兩個數組，並返回較小值的數組元素。如果不指定軸參數，則返回所有元素的最小值。

axis參數可以接受一個整數值或一個元組。如果傳遞的是一個整數值，則指定按照該軸的最小值計算。如果指定的是一個元組，則可以指定多個軸。

import numpy as np

arr2d = np.array([[1, 2], [3, 4], [5, 6]])

# 沿第一列計算最小值
print(np.amin(arr2d, axis=0))

# 沿第一行計算最小值
print(np.amin(arr2d, axis=1))

# 將所有元素與標量值比較
print(np.amin(arr2d, axis=None))

四、nan和inf數組

當數組中存在nan或inf時，如果不指定軸參數，則np.amin()函數會將整個數組視為nan或inf。

import numpy as np

# 數組中存在nan值
arr_nan = np.array([1, np.nan, 2, 3])
print(np.amin(arr_nan))

# 數組中存在inf值
arr_inf = np.array([1, 2, 3, np.inf])
print(np.amin(arr_inf))

如果指定了軸參數，則忽略nan或inf值並返回最小值。

import numpy as np

# 數組中存在nan值
arr2d_nan = np.array([[1, 2], [3, np.nan], [4, 5]])
print(np.amin(arr2d_nan, axis=0))

# 數組中存在inf值
arr2d_inf = np.array([[1, 2], [3, 4], [5, np.inf]])
print(np.amin(arr2d_inf, axis=1))

五、總結

本文中，我們介紹了NumPy中的amin()函數，並詳細討論了它的用法、返回值類型、軸參數以及處理nan和inf數組的方法。amin()函數是NumPy中用於查找數組最小值的有用函數。

要使用np.amin()函數，需要首先安裝NumPy庫，並發現如何使用jar包，或者使用pip命令來安裝NumPy庫。