Numpy 将函数应用于数组 [英] Numpy apply function to array

问题描述

例如，我有函数:

f1 = lambda x: x % 2

如果我想修改array = np.linspace(0, 5, 6) 我可以做f1(array).一切都按预期进行:

If I want to modify array = np.linspace(0, 5, 6) I can do f1(array). Everything works as expected:

[0. 1. 0. 1. 0. 1.]

如果我将函数更改为:

f2 = lambda x: 0
print(f2(array))

给了我 0 而我期望 [0.0. 0. 0. 0. 0.].如何实现一致性?

gives me 0 while I expected [0. 0. 0. 0. 0. 0.]. How to achieve consistency?

推荐答案

您可以使用下面的代码来实现理想的输出

You can use below code to achieve desirable output

import numpy as np
array = np.linspace(0, 5, 6)
f2 = lambda x: x-x
print(f2(array))

这篇关于Numpy 将函数应用于数组的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！