Question

我有以下两个n个元素的numpy数组：

A = np.array([2 5 8 9 8 7 5 6])
B = np.array([8 9 6 5 2 8 5 7])

我想获得数组C：

C = np.array([sqrt(2^2+8^2) sqrt(5^2+9^2) ... sqrt(6^2+7^2)])

即，数组C由n个元素组成;每个元素将等于A中相应元素的平方的平方根加上B中相应元素的平方。

我尝试使用np.apply_along_axis，但似乎此功能仅适用于一个数组。

Answer 1

如评论中所述，您可以使用：

C = np.sqrt(A**2 + B**2)

或者您可以使用comprehension和zip：

C = [sqrt(a**2 + b**2) for a, b in zip(A,B)]

Answer 2

如果您的阵列规模很大，请考虑使用np.square代替**运算符。

In [16]: np.sqrt(np.square(A) + np.square(B))
Out[16]: 
array([  8.24621125,  10.29563014,  10.        ,  10.29563014,
         8.24621125,  10.63014581,   7.07106781,   9.21954446])

但执行时间的差异非常小。

In [13]: ar = np.arange(100000)

In [14]: %timeit np.square(ar)
10000 loops, best of 3: 158 µs per loop

In [15]: %timeit ar**2
10000 loops, best of 3: 179 µs per loop

如何在两个不同的数组上从元素计算返回数组？

2 个答案: