python 有效地将 function 应用于多个 arrays

Question

（python 的新手，如果这个问题很基础，我深表歉意）

假设我创建了一个 function 来计算一些方程式

def plot_ev(accuracy,tranChance,numChoices,reward): 
     ev=(reward-numChoices)*1-np.power((1-accuracy),numChoices)*tranChance)
     return ev

accuracy、tranChance 和 numChoices 都是浮点数 arrays

e.g. 
accuracy=np.array([.6,.7,.8])
tranChance=np.array([.6,.7,8])
numChoices=np.array([2,.3,4])

我将如何在我的 3 arrays 上运行和 plot plot_ev 以便我最终得到一个具有所有元素组合的 output（理想情况下不运行 3 forloops）

理想情况下，我会有一个 plot 显示所有组合的 output（第一个元素来自 accuracy，所有元素来自 transChance 和 numChoices，第二个元素来自 accuracy，所有元素来自 transChance 和 numChoices 等等）

提前致谢！

Answer 1

使用numpy.meshgrid制作三个变量值的所有组合的数组。

products = np.array(np.meshgrid(accuracy, tranChance, numChoices)).T.reshape(-1, 3)

然后再次转置它并提取三个更长的 arrays 以及每个组合中三个变量的值：

accuracy_, tranChance_, numChoices_ = products.T

您的 function 仅包含可以在 numpy arrays 上执行的操作，因此您可以简单地将这些 arrays 作为参数输入 function：

reward = ??  # you need to set the reward value
results = plot_ev(accuracy_, tranChance_, numChoices_, reward)

或者考虑使用 pandas dataframe，这将提供更清晰的列标签。

import pandas as pd
df = pd.DataFrame(products, columns=["accuracy", "tranChance", "numChoices"])
df["ev"] = plot_ev(df["accuracy"], df["tranChance"], df["numChoices"], reward)