python画图设置彩色线条_更改绘图中的线条颜色

我在Python中用一些数据绘制了一个曲面图。

现在我试图改变这种情节的风格。但不幸的是，我陷入了线颜色的困境。它的默认值是黑色，但我想让它变成红色或其他颜色。

我的代码是：from mpl_toolkits.mplot3d import Axes3D

import matplotlib.pyplot as plt

from matplotlib import cm

import numpy as np

data=np.loadtxt("test.txt")

def formateU(data):

U = np.zeros((20,20))

for value in data:

U[value[0],value[1]] = value[2]

return U

U = formateU(data)

y,x=np.meshgrid(np.linspace(0.,19,20),np.linspace(0.,19,20))

fig = plt.figure()

ax=plt.axes(projection='3d')

ax.plot_surface(x,y,U,rstride=1,cstride=1,alpha=0,linewidth=0.5)

ax.view_init(30, 45)

plt.savefig("test.png")

plt.show()

很明显，这是一个附加的论点：ax.plot_surface(x,y,U,rstride=1,cstride=1,alpha=0,linewidth=0.5)

但我想不通。

你能帮帮我吗？