Python 遍历numpy数组/矩阵并将其存储在单独的数组中

Python 遍历numpy数组/矩阵并将其存储在单独的数组中,python,numpy,for-loop,Python,Numpy,For Loop,我有一个矩阵/numpy数组a,需要对所有元素执行函数f,然后将结果存储在矩阵B中。我该怎么做,我正在考虑在继续时附加一个空数组B,但这是最好的方法吗?您可以使用: 输出 print(B) array([[1, 2, 3], [4, 5, 6], [7, 8, 9]]) 大多数操作可以直接在阵列本身上完成。例如,如果您想向数组的每个元素添加一个元素,只需调用 arr2 = arr1+1 如果所需的操作不是通用的,则可能需要进行矢量化,但这将以牺牲执行时间为代价。

我有一个矩阵/numpy数组a,需要对所有元素执行函数f,然后将结果存储在矩阵B中。我该怎么做,我正在考虑在继续时附加一个空数组B,但这是最好的方法吗?

您可以使用:

输出

print(B)
array([[1, 2, 3],
       [4, 5, 6],
       [7, 8, 9]])


大多数操作可以直接在阵列本身上完成。例如,如果您想向数组的每个元素添加一个元素,只需调用

arr2 = arr1+1

如果所需的操作不是通用的,则可能需要进行矢量化,但这将以牺牲执行时间为代价。对函数进行矢量化仍然会导致循环操作,因为通用函数(如上所述)是作为标量执行的。

这是否回答了您的问题?告诉我们有关
A
-形状和数据类型的信息。还有关于
f
。它接受什么样的输入?如果函数只接受标量值,那么很难在1d列表上击败列表理解。如果有人告诉你应该使用
numpy
,那就花点时间来学习它的基础知识。糟糕的
numpy
不会为您节省时间。
arr2 = arr1+1