我有一个 numpy 数组
[
[1,5,6],
[10,12,20]
]
我想对数组的每个元素进行操作,获取当前元素值并执行操作。一种类似 np.square 函数的批量操作。
即 x = (x + 1) * 2 结果将是:
[
[4,12,14],
[22,26,42]
]
我知道我可以为每个元素执行一个 for 循环并应用该操作,但我想要更紧凑的语法。
您可以使用 numpy:
import numpy as np
lst = [
[1, 5, 6],
[10, 12, 20]
]
lst = np.array(lst)
lst = (lst + 1) * 2
print(lst)
根据需要输出:
[[ 4 12 14]
[22 26 42]]
如果最后你希望它成为一个列表,你可以将它转换回来,但np
数组很棒。
使用 numpy 向量化的解决方案:
# define numpy array
arr = np.array([
[1,5,6],
[10,12,20]
])
# create function
func = np.vectorize(lambda x: (x + 1) * 2)
# apply function to array
func(arr)