[Python知识库]Python散点图加入颜色维度

简单几行代码就ok，稍微记录一下

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

aaa=pd.read_csv('lq.csv')
I=aaa['Imax [A]']
plt.scatter(aaa['Thet []'],aaa['avg(L_q) []'],c=I)
plt.xlabel('Thet(°)')
plt.ylabel('lq(mH)')
plt.colorbar(label='Imax(A)',orientation="horizontal")
plt.savefig('lq',bbox_inches = 'tight',pad_inches = 0,dpi =200)

其中，那个colorbar是设置下面那个颜色条的，因为还要加注释，我就把它横过来了。

?