在 numpy 數組上映射函數的最有效方法

| | | | | | | | | | | | |

將函數映射到 numpy 數組的最有效方法是什麼?我在當前項目中一直這樣做的方式如下:

import numpy as np x = np.array([1, 2, 3, 4, 5]) # 獲取x 中每個元素的平方數組 squarer = lambda t: t ** 2 squares = np.array([squarer(xi) for xi in x]) 

但是,這看起來像這可能效率很低,因為我使用列表推導將新數組構造為 Python 列表,然後再將其轉換回 numpy 數組。

我們能做得更好嗎?