np.frompyfunc()函数的使用 Numpy | 函数向量化

python自定义函数并调用:

def func_x(a):
	y = a + 1
	return y

print(func_x(1)) # 调用
# 结果:
# 2

但是如果a不是一个数,而是一个向量/数组呢?这时候就需要借助numpy中的通用函数np.frompyfunc(func, 1, 1)
这里的参数func指的是你要使用的函数,第二个参数为func中的参数个数,第三个参数为func中的返回值的个数
假设A = [0, 1, 2]

def func_x(a):
	y = a + 1
	return y
A = [0, 1, 2]
func_ = np.frompyfunc(func_x, 1, 1)
y = func_(A)
print(y)
# 结果:
# [1 2 3]

参考博客:函数向量化运算的多种方法function/lambda + numpy.frompyfunc()/map和numpy.vectorize()

全部评论

相关推荐

06-07 21:26
江南大学 C++
话不多说,直接上时间线和图片1.2024年10月底发offer,并签三方2.2025年5月底公司违约
从零开始的转码生活:希望所有签了三方但直接违约的公司都倒闭!都倒闭!都倒闭!
点赞 评论 收藏
分享
能干的三文鱼刷了100道题:公司可能有弄嵌入式需要会画pcb的需求,而且pcb能快速直观看出一个人某方面的实力。看看是否有面试资格。问你问题也能ai出来,pcb这东西能作假概率不高
点赞 评论 收藏
分享
完美的潜伏者许愿简历通过:我上表jd,请求封我做后端大将军的事,北京有消息了:竟然不许!!! 他们一定是看我没有实习,这才故意驳回我的请求!
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务