蟒蛇 - 散點圖日期和第三個變量作爲顏色

Question

我想繪製一個x-y圖，用x或y作爲日期變量，並使用第三個變量來着色點。 我設法做到這一點，如果沒有3個變量是日期，使用：

ax.scatter(df['x'],df['y'],s=20,c=df['z'], marker = 'o', cmap = cm.jet) 

搜索後，我發現，正常的陰謀，我們必須使用plot_date（）。不幸的是，我沒有能夠着色點。 任何人都可以幫助我嗎？

這裏是一個小例子：

import matplotlib, datetime 
import matplotlib.pyplot as plt 
import matplotlib.cm as cm 
import pandas as pd 

todayTime=datetime.datetime.now(); 
df = pd.DataFrame({'x': [todayTime+datetime.timedelta(hours=i) for i in range(10)], 'y': range(10),'z' : [2*j for j in range(10)]}); 
xAlt=[0.5*i for i in range(10)]; 

fig, ax = plt.subplots() 
ax.scatter(df['x'],df['y'],s=20,c=df['z'], marker = 'o', cmap = cm.jet) 
plt.show() 

您可以替換DF [ 'X']由xAlt看到期望的結果

謝謝

Answer 1

據我所知，一個必須使用scatter才能按照您描述的方式着色點。一種解決方法是使用FuncFormatter將刻度標籤轉換爲x軸上的時間。下面的代碼將日期轉換爲數字，繪製散點圖，並使用FuncFormatter將刻度標籤轉換回日期。

import matplotlib, datetime 
import matplotlib.pyplot as plt 
import matplotlib.cm as cm 
import pandas as pd 
from matplotlib.ticker import FuncFormatter 

todayTime=datetime.datetime.now() 
df = pd.DataFrame({'x': [todayTime+datetime.timedelta(hours=i) for i in range(10)], 'y': range(10),'z' : [2*j for j in range(10)]}) 

def my_formatter(x, pos=None): 
    d = matplotlib.dates.num2date(x) 
    if len(str(d.minute)) == 1: 
     mn = '0{}'.format(d.minute) 
    else: 
     mn = str(d.minute) 
    if len(str(d.hour)) == 1: 
     hr = '0{}'.format(d.hour) 
    else: 
     hr = str(d.hour) 
    return hr+':'+mn 

major_formatter=FuncFormatter(my_formatter) 

nums = np.array([matplotlib.dates.date2num(di) for di in df['x']]) 

fig, ax = plt.subplots() 
ax.xaxis.set_major_formatter(major_formatter) 
ax.scatter(nums,df['y'],s=20,c=df['z'], marker = 'o', cmap = cm.jet) 
xmin = df['x'][0]-datetime.timedelta(hours=1) 
xmax = df['x'][len(df['x'])-1]+datetime.timedelta(hours=1) 
ax.set_xlim((xmin,xmax)) 
plt.show()