您好, 欢迎来到 !    登录 | 注册 | | 设为首页 | 收藏本站

如何在python中计算外部函数?

如何在python中计算外部函数?

只要确保以正确广播的方式编写函数,就可以

f(x_arr[:, None], y_arr)

将其应用到所有对的元素的两个1维阵列x_arry_arr

例如,要以广播的方式编写示例函数,应将其编写为

def func(x, y):
    return np.cos(x) + np.sin(y) + x*y

因为np.cosnp.sin+,和*跨阵列广播和量化。

至于它不广播?好吧,也许有人建议这样做np.vectorize,但是您需要牢记很多棘手的事情,例如保持一致的输出dtype且没有副作用。如果您的函数不广播,我建议您仅使用列表推导:

np.array([[f(xval, yval) for yval in y_arr] for xval in x_arr])
python 2022/1/1 18:51:58 有327人围观

撰写回答


你尚未登录,登录后可以

和开发者交流问题的细节

关注并接收问题和回答的更新提醒

参与内容的编辑和改进,让解决方法与时俱进

请先登录

推荐问题


联系我
置顶