如何在matplotlib plot_trisurf中为单个元素着色

import matplotlib.pyplot as plt
import numpy as np

# 生成三角形的顶点坐标
x = np.array([0, 1, 2])
y = np.array([0, 1, 0])
z = np.array([0, 0, 1])

# 生成三角形的连接关系
triangles = np.array([[0, 1, 2]])

# 生成每个三角形的颜色
colors = np.array([[1, 0, 0]])  # 这里只给第一个三角形着红色

# 绘制三维曲面图
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
ax.plot_trisurf(x, y, z, triangles=triangles, facecolors=colors)

# 显示图形
plt.show()