根据分类数据绘制日期时间(Y轴)

时间:2019-08-02 15:55:07

标签: python matplotlib categorical-data

我正在尝试针对datetime中的一系列matplotlib值绘制分类信息。如果分类数据表示为字符串,则可以使绘图起作用。但是,我希望对Y轴进行分类,以便可以按正确的顺序对其进行排序。

以下代码段显示了到目前为止的内容。在图中,将y替换为y_cat,而matplotlib会引发错误:

import pandas as pd
import numpy as np
import calendar, datetime

import matplotlib as mpl
import matplotlib.pyplot as plt

# %matplotlib inline #for Jupyter notebooks

x = pd.date_range('2015/08/01', freq='4M', periods=9)
y = pd.Series(['Good', 'Very Good', 'Very Good', 'Average', 'Average', 'Good', 'Excellent', 'Excellent', 'Excellent'])
y_cat = pd.Categorical(y, categories=['Poor', 'Average', 'Good', 'Very Good', 'Excellent'], ordered=True)

fig, currAX = plt.subplots(figsize=(10, 4))
label_format = {'fontsize':12, 'fontweight':'bold'}
title_format = {'fontsize':15, 'fontweight':'bold'}

currAX.plot(x, y, color='crimson', linestyle='-')
#uncomment for error
#currAX.plot(x, y_cat, color='crimson', linestyle='-')

currAX.xaxis.set_major_formatter(mpl.dates.DateFormatter('%Y %b'))

currAX.spines['top'].set_visible(False)
currAX.spines['right'].set_visible(False)
currAX.spines['left'].set_visible(False)

currAX.set_xlabel('Review Period', **label_format)
currAX.set_ylabel('Review Rating', **label_format)

fig.tight_layout()
plt.show();

### ERROR:
IndexError: tuple index out of range

我想在Y轴上看到带有评论类别的图表,按从上到下,从上到下的顺序排序

1 个答案:

答案 0 :(得分:0)

您期望list时将它们作为tuple传递。此处代码已更正:

已编辑:如果您还希望在Y轴上具有有序的值,则需要指定一个数字值,以便绘图知道每个点的放置位置。然后将INT值替换为标签。这里是更新的代码。

import pandas as pd
import numpy as np
import calendar, datetime

import matplotlib as mpl
import matplotlib.pyplot as plt

# %matplotlib inline #for Jupyter notebooks


x = pd.date_range('2015/08/01', freq='4M', periods=9).tolist()
y = pd.Series(['Good', 'Very Good', 'Very Good', 'Average', 'Average', 'Good', 'Excellent', 'Poor', 'Excellent']).tolist()


### create a conversion DICT
conversion = { \
        "Poor" : 0, \
        "Average" : 1, \
        "Good" : 2, \
        "Very Good" : 3, \
        "Excellent" : 4 \
}
## open a list and insert in it the INT corresponding value
y_converted = []
for v in y :
    y_converted.append(conversion[v])

fig, currAX = plt.subplots(figsize=(10, 4))
label_format = {'fontsize':12, 'fontweight':'bold'}
title_format = {'fontsize':15, 'fontweight':'bold'}

### pass as tuple
currAX.plot(x, y_converted, color='crimson', linestyle='-')


currAX.xaxis.set_major_formatter(mpl.dates.DateFormatter('%Y %b'))

currAX.spines['top'].set_visible(False)
currAX.spines['right'].set_visible(False)
currAX.spines['left'].set_visible(False)

currAX.set_xlabel('Review Period', **label_format)
currAX.set_ylabel('Review Rating', **label_format)

### tell matplotlib the ticks and labels to use on Y-axis
currAX.set_yticks( list(conversion.values()) )
currAX.set_yticklabels( list(conversion.keys()) )

fig.tight_layout()
plt.show();

结果: enter image description here