修改代码以允许在x中输入多个条目

Question

是否可以修改以下功能的代码以便允许在x中输入多个条目（即，我想一次查询多个点）？

def banana(x): 
    return exp(((-x[0]**2/200))-0.5*(x[1]+0.05*(x[0]**2) - 100*0.05)**2) 

Answer 1

而是改变功能的，可以考虑使用内置map功能的函数应用于集合中的所有元素。在此，banana的实现保持简单。以下是如何使用map函数的示例。

entries = [...] 
mapped_entries = map(banana, entries) 

Answer 2

使用numpy。看起来你的功能使用了一对数字（x[0]和x[1]）。

import numpy as np 

def apple(x): # Note that `x` is an array, not a tuple pair. 
    return np.exp((((-x[0]**2/200)) - 0.5 * (x[1] + 0.05 * x[0]**2 - 100 * 0.05) ** 2)) 

x = np.array([[1, 2, 3], [4, 5, 6]]) 
>>> apple(x) 
array([ 0.23427726, 0.36059494, 0.12857409])