📅  最后修改于: 2022-03-11 14:45:05.334000             🧑  作者: Mango
import numpy as np
arr = np.array([1,2,3,4])
print(np.apply_along_axis(lambda x : x ** 2, 0, arr))
#Output: array([ 1, 4, 9, 16])