在做数据分析的时候,需要根据条件将不同的数值设置为不同的颜色,然后画出来。网上找了半天没找到合适的解决方案,自己研究了一个方案,供大家批评指正
首先导入必要的库
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函数设置你的条件