matplotlib根据条件设置标签颜色

在做数据分析的时候，需要根据条件将不同的数值设置为不同的颜色，然后画出来。网上找了半天没找到合适的解决方案，自己研究了一个方案，供大家批评指正

首先导入必要的库

import numpy as np
import matplotlib.pyplot as plt
import pandas as pd

然后导入excel中的数据，这里略过

接下来在画图的时候，我想将60分以上的点标注为绿色（通过），60分以下的点标注为红色（未通过）

for a,b in zip(dataset.iloc[:,i],dataset.iloc[:,-2]):
        if b < 60:
            plt.scatter(a,b,color='red')
        else:
            plt.scatter(a,b,color='green')

这里要注意：

1.通过pandas读入的excel数据为dataframe格式，用iloc切片成list

2.使用zip函数将list数据组成元组，利用for循环遍历元组内容

3.用if函数设置你的条件