錯誤想要繪製在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中的基本類型，這樣做會導致很多麻煩。