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

A cranky quirk:将函数应用于所有两个1D数组对,以获得一个2D数组

A cranky quirk:将函数应用于所有两个1D数组对,以获得一个2D数组

您可以使用numpy广播在两个数组上进行计算,使用以下命令将其a转换为垂直2D数组newaxis

In [11]: a = np.array([1, 2, 3]) # n1 = 3
    ...: b = np.array([4, 5]) # n2 = 2
    ...: #if function is c(i, j) = a(i) + b(j)*2:
    ...: c = a[:, None] + b*2

In [12]: c
Out[12]: 
array([[ 9, 11],
       [10, 12],
       [11, 13]])

进行基准测试:

In [28]: a = arange(100)

In [29]: b = arange(222)

In [30]: timeit r = np.array([[f(i, j) for j in b] for i in a])
10 loops, best of 3: 29.9 ms per loop

In [31]: timeit c = a[:, None] + b*2
10000 loops, best of 3: 71.6 us per loop
其他 2022/1/1 18:47:32 有460人围观

撰写回答


你尚未登录,登录后可以

和开发者交流问题的细节

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

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

请先登录

推荐问题


联系我
置顶