matplotlib.pyplot按y值大小设置颜色

可以使用`plt.scatter`函数按照y值的大小设置颜色，具体步骤如下：

1. 生成x和y的数据：

import numpy as np
x = np.random.rand(50)
y = np.random.rand(50)

2. 设置颜色映射：

import matplotlib.pyplot as plt
from matplotlib.colors import ListedColormap
cmap = ListedColormap(['blue', 'green', 'red'])

这里使用了一个三种颜色的颜色映射，y值小的点对应蓝色，y值中等的点对应绿色，y值大的点对应红色。

3. 绘制散点图并设置颜色：

plt.scatter(x, y, c=y, cmap=cmap)

其中，`c=y`表示将y的值作为颜色参数传递给`plt.scatter`函数，`cmap=cmap`表示使用上一步设置的颜色映射。

完整代码如下：

import numpy as np
import matplotlib.pyplot as plt
from matplotlib.colors import ListedColormap

x = np.random.rand(50)
y = np.random.rand(50)

cmap = ListedColormap(['blue', 'green', 'red'])

plt.scatter(x, y, c=y, cmap=cmap)

plt.show()