我正在尝试针对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轴上看到带有评论类别的图表,按从上到下,从上到下的顺序排序
答案 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();