Question

# -*- coding: utf-8 -*-
import matplotlib.pyplot as plt 
import numpy as np 
import math


#task 2e

x = np.linspace(-0.0001,0.1,50) 

#constants
e0=8.85*10 ** (-12)

r0=3 * 10 ** (-3)
Q=2 * 10** (-9)

Q2=10 * 10*(-9)
r2=5*10**(-3)


v=(Q/((4*math.pi*e0)*(math.sqrt((x**2+r0**2)))))


v2=v+(Q2/((4*math.pi*e0)*(math.sqrt(((x-2)**2+r2**2)))))

plt.plot(x, v)
plt.plot(x, v2)
plt.xlabel("Meter")
plt.ylabel("V1/2(x)")

运行此代码会产生以下TypeError：

TypeError：只能将length-1数组转换为Python标量 21 v =（Q /（（4 * math.pi * e0）（math.sqrt（（x * 2 + r0 ** 2））））））

Answer 1

使用numpy.sqrt而不是math.sqrt。 numpy.sqrt期望标量或数组作为输入，另一方面math.sqrt只能处理标量。

>>> import numpy as np
>>> import math
>>> a = np.arange(5)
>>> np.sqrt(a)
array([ 0.        ,  1.        ,  1.41421356,  1.73205081,  2.        ])
#error
>>> math.sqrt(a)
Traceback (most recent call last):
  File "<ipython-input-78-c7d50051514f>", line 1, in <module>
    math.sqrt(a)
TypeError: only length-1 arrays can be converted to Python scalars

>>>

Answer 2

使用 np 代替 math.sqrt

v=(Q/((4*math.pi*e0)*(np.sqrt((x**2+r0**2)))))


v2=v+(Q2/((4*math.pi*e0)*(np.sqrt(((x-2)**2+r2**2)))))

TypeError：只有length-1数组可以使用NUMPY转换为Python标量

2 个答案: