在python中绘制一个常量

Question

我有两个数组，其中C是常量

import numpy as np
import matplotlib.pyplot as plt

constants = var('C')
x = [0, 1, 2, 3, 4, 5]
y = [C, 2*C, 3*C, 4*C, 5*C]

fig = plt.figure()
plt.plot(x, y)
plt.grid(true)
plt.savefig("pru")
plt.close()

但是当我尝试使用Matpotlib进行绘图时，请给出我的错误：

TypeError: unable to simplify to float approximation

我如何策划这个？

Answer 1

如果使用numpy数组，则将数组乘以常量将按元素方式自动完成：

import numpy as np
import pylab as plt

c = 2.0
x = np.arange(5)
y = c*x

plt.plot(x,y)
plt.show()