matplotlib 求助：如何给颜色映射配置图例

我写的代码是这样的，但是无法隐射到颜色，我想要图例中的颜色是与之相对应的。

from sklearn.datasets import load_iris
iris = load_iris()
features = iris.data.T
plt.scatter(features[0], features[1], alpha=0.2,
s=100*features[3], c=iris.target, cmap='viridis')
plt.xlabel(iris.feature_names[0])
plt.ylabel(iris.feature_names[1]);


count = 0
for t in set(iris.target):
    plt.scatter([],[] ,c=[], cmap='viridis', label=iris.target_names[count])
    count += 1

plt.legend(scatterpoints=1)

注：原来的代码是没有图例的

from sklearn.datasets import load_iris
iris = load_iris()
features = iris.data.T
plt.scatter(features[0], features[1], alpha=0.2,
s=100*features[3], c=iris.target, cmap='viridis')
plt.xlabel(iris.feature_names[0])
plt.ylabel(iris.feature_names[1]);

 图片是这样的