错误想要绘制在python一个简单的函数

Question

ValueError异常：阵列的具有多于一个元素的真值是不明确的。使用a.any（）或a.all（）

我得到取决于积分方法不同的错误。该功能在给定单个值时正常工作。

import matplotlib.pyplot as plt 
import scipy as sp 
import numpy as np 


def mass_enc(R): 
    def int(r): return r**2 * r 
    return sp.integrate.quad(int, 0, R) 

print(mass_enc(10)) 

t1 = np.arange(0.1, 5.0, 0.1) 
plt.plot(t1, mass_enc(t1)) 

Answer 1

问题是，您打电话给sp.integrate.quad并将数组作为参数。虽然一些功能实际上允许，但是quad不允许。所以你需要分别提供每个值R。这可以通过map(function, iterable)完成。所以这里是你如何做到的。

import matplotlib.pyplot as plt 
import scipy as sp 
import numpy as np 

def inte(r): 
    return r**2 * r 

def mass_enc(R): 
    return sp.integrate.quad(inte, 0, R)[0] 

print(mass_enc(10)) 

t1 = np.arange(0.1, 5.0, 0.1) 
m = map(mass_enc, t1) 
plt.plot(t1, m) 
plt.show() 

记住，你应该从未呼叫蟒蛇int任何对象，因为int是Python中的基本类型，这样做会导致很多麻烦。