要在任意轴上操作，可以使用NumPy库中的函数。下面是一个示例代码，展示了如何编写一个函数来在任意轴上对一个二维数组进行操作：

import numpy as np

def operate_on_axis(arr, operation, axis=0):
    if axis == 0:
        return operation(arr, axis=0)
    elif axis == 1:
        return operation(arr, axis=1)
    else:
        raise ValueError("Invalid axis value. Axis must be 0 or 1.")

# 示例操作函数：计算每行或每列的平均值
def calculate_mean(arr, axis):
    return np.mean(arr, axis=axis)

# 示例二维数组
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

# 在第0轴上计算平均值 (按行计算)
result = operate_on_axis(arr, calculate_mean, axis=0)
print("按行计算平均值：", result)

# 在第1轴上计算平均值 (按列计算)
result = operate_on_axis(arr, calculate_mean, axis=1)
print("按列计算平均值：", result)

输出：

按行计算平均值： [4. 5. 6.]
按列计算平均值： [2. 5. 8.]

在该示例中，operate_on_axis函数接受一个二维数组，一个操作函数以及一个轴参数。它根据轴参数的值来决定是在第0轴（按行）还是第1轴（按列）上进行操作。calculate_mean函数是一个示例操作函数，用于计算每行或每列的平均值。通过调用operate_on_axis函数并传递适当的参数，可以在任意轴上执行操作。