如何用Numpy聚合方法替代a[-1]获取一维数组最后元素？

Numpy原生并没有提供np.last()或np.first()这类直接返回数组首尾元素的聚合方法，但可以通过以下方式实现需求，让你的operate_on_array函数能支持这类操作：

方法一：自定义函数并挂载到Numpy模块 #

直接写好获取首尾元素的函数，然后把它们挂载到np模块上，这样就能完全复用你现有的operate_on_array函数逻辑，不需要修改原函数代码：

import numpy as np

# 定义获取最后一个元素的函数
def np_last(a: np.ndarray):
    return a[-1]

# 定义获取第一个元素的函数
def np_first(a: np.ndarray):
    return a[0]

# 将自定义函数挂载到numpy模块
np.last = np_last
np.first = np_first

# 你的原有函数无需改动
def operate_on_array(a: np.array, np_method_name: str):
     method = getattr(np, np_method_name)
     return method(a)

# 测试示例
test_arr = np.array([10, 20, 30, 40, 50])
print(operate_on_array(test_arr, "last"))  # 输出 50
print(operate_on_array(test_arr, "first")) # 输出 10

方法二：重构自定义函数以支持直接传入方法 #

如果不想修改Numpy模块的属性，也可以稍微调整operate_on_array函数，让它既支持传入Numpy原生方法名，也能直接传入自定义函数：

import numpy as np

def np_last(a: np.ndarray):
    return a[-1]

def np_first(a: np.ndarray):
    return a[0]

# 扩展后的函数
def operate_on_array(a: np.array, method):
    # 如果传入的是字符串，就从numpy模块获取方法；否则直接使用传入的函数
    if isinstance(method, str):
        method = getattr(np, method)
    return method(a)

# 测试示例
test_arr = np.array([10, 20, 30, 40, 50])
print(operate_on_array(test_arr, np_last))   # 输出 50
print(operate_on_array(test_arr, "mean"))    # 兼容原有Numpy方法，输出30

内容的提问来源于stack exchange，提问作者Pedro Fernández Soler