python - Matplotlib 中多个散点图的彩色标记日期

Question

我有 3 个数据框，它们有 3 个我想映射到散点图中的特征：

Date  Energy  Cost

我需要创建一系列散点图，绘制能源和成本，但在日期列中按年份进行颜色编码。

日期采用日期时间格式，持续 4 年，如下所示：

2015-12-30 00:00:00
2016-20-12 00:00:00
etc...

我知道我可以绘图plt.scatter(data[:], data[:].... c=data[:])

但是无法弄清楚如何专门针对时间间隔正确格式化它，以使每种颜色在 4 年中的每一年都不同。

标签： pythonpython-3.xpandasmatplotlibplot

Answer 1

用于map创建可以作为颜色传递的系列。

MCVE

import pandas as pd
import numpy as np

np.random.seed(123)
dates = pd.date_range('2014-01-01', '2017-12-31', freq='12D')
df = pd.DataFrame({'Date': dates,
                   'Energy': np.random.randint(1, 1000, len(dates)),
                   'Cost': np.random.randint(10000, 15000, len(dates))})

代码

color_d = {2014: 'red', 2015: 'blue', 2016: 'orange', 2017: 'black'}
df.plot.scatter(x='Energy', y='Cost', c=df.Date.dt.year.map(color_d))

---

或者使用 matplotlib

import matplotlib.pyplot as plt

plt.scatter(df.Energy.values, df.Cost.values, c=df.Date.dt.year.map(color_d))
_ = plt.xlabel('Energy')
_ = plt.ylabel('Cost')
plt.show()