10

我创建了一个看起来像的情节在此处输入图像描述

我有几个问题:

  1. 我怎样才能特别显示周末。我曾想过的一些方法是获取与周末相对应的索引,然后在 xlims 之间绘制透明条。也可以绘制相同的矩形。如果可以在 Pandas 中简单地完成,那将是最好的。
  2. 日期格式不是最漂亮的

以下是用于生成此图的代码

ax4=df4.plot(kind='bar',stacked=True,title='Mains 1 Breakdown');
ax4.set_ylabel('Power (W)');
idx_weekend=df4.index[df4.index.dayofweek>=5]
ax.bar(idx_weekend.to_datetime(),[1800 for x in range(10)])

专门用于突出显示ax.bar周末,但它不会产生任何可见的输出。(问题 1)对于问题 2,我尝试使用 Major Formatter 和 Locators,代码如下:

ax4=df4.plot(kind='bar',stacked=True,title='Mains 1 Breakdown');
ax4.set_ylabel('Power (W)');
formatter=matplotlib.dates.DateFormatter('%d-%b');
locator=matplotlib.dates.DayLocator(interval=1);
ax4.xaxis.set_major_formatter(formatter);
ax4.xaxis.set_major_locator(locator);

产生的输出如下: 在此处输入图像描述

了解 Dataframe 的样子可能会有所帮助

In [122]:df4

Out[122]:
<class 'pandas.core.frame.DataFrame'>
DatetimeIndex: 36 entries, 2011-04-19 00:00:00 to 2011-05-24 00:00:00
Data columns:
(0 to 6 AM) Dawn          19  non-null values
(12 to 6 PM) Dusk         19  non-null values
(6 to 12 Noon) Morning    19  non-null values
(6PM to 12 Noon) Night    20  non-null values
dtypes: float64(4)
4

2 回答 2

13

我尝试了很多,现在这些技巧有效。等待更 Pythonic 和一致的解决方案。标签问题的解决方案:

def correct_labels(ax):
    labels = [item.get_text() for item in ax.get_xticklabels()]
    days=[label.split(" ")[0] for label in labels]
    months=["Jan","Feb","Mar","Apr","May","Jun","Jul","Aug","Sep","Oct","Nov","Dec"]
    final_labels=[]
    for i in range(len(days)):
        a=days[i].split("-")
        final_labels.append(a[2]+"\n"+months[int(a[1])-1])
    ax.set_xticklabels(final_labels)

另外,在绘图时,我进行了以下更改

ax=df.plot(kind='bar',rot=0)

这使得标签在 0 旋转。

为了找到周末并突出显示它们,我编写了以下两个函数:

def find_weekend_indices(datetime_array):
    indices=[]
    for i in range(len(datetime_array)):
        if datetime_array[i].weekday()>=5:
            indices.append(i)
    return indices

def highlight_weekend(weekend_indices,ax):
    i=0
    while i<len(weekend_indices):
         ax.axvspan(weekend_indices[i], weekend_indices[i]+2, facecolor='green', edgecolor='none', alpha=.2)
         i+=2

现在,该图看起来更有用并涵盖了这些用例。在此处输入图像描述

于 2013-05-18T12:27:01.753 回答
6

现在 Pandas.dt在每个系列中都支持强大的命名空间,无需任何明确的 Python 循环就可以识别每个周末的开始和结束。只需过滤您的时间值t.dt.dayofweek >= 5以仅选择周末的时间,然后按每周不同的虚构值进行分组——我在这里使用它是year * 100 + weekofyear因为结果看起来201603很容易阅读以进行调试。

结果函数是:

def highlight_weekends(ax, timeseries):
    d = timeseries.dt
    ranges = timeseries[d.dayofweek >= 5].groupby(d.year * 100 + d.weekofyear).agg(['min', 'max'])
    for i, tmin, tmax in ranges.itertuples():
        ax.axvspan(tmin, tmax, facecolor='orange', edgecolor='none', alpha=0.1)

只需将轴和作为您的轴的时间序列传递给它x,它就会为您突出显示周末!

于 2016-02-05T17:37:09.430 回答