def ibar(indicator, new=True, axes=None,
legend_on=False, text_on=False, text_color='k', label=None,
width=0.4, color='r', edgecolor='r',
zero_on=False, *args, **kwargs):
"""绘制indicator柱状图
:param Indicator indicator: Indicator实例
:param axes: 指定的坐标轴
:param new: 是否在新窗口中显示，只在没有指定axes时生效
:param legend_on: 是否打开图例
:param text_on: 是否在左上角显示指标名称及其参数
:param text_color: 指标名称解释文字的颜色，默认为黑色
:param str label: label显示文字信息，text_on 及 legend_on 为 True 时生效
:param zero_on: 是否需要在y=0轴上绘制一条直线
:param width: Bar的宽度
:param color: Bar的颜色
:param edgecolor: Bar边缘颜色
:param args: pylab plot参数
:param kwargs: pylab plot参数
"""
if not indicator:
print("indicator is None")
return
if not axes:
axes = create_figure() if new else gca()
if not label:
label = "%s %.2f" % (indicator.long_name, indicator[-1])
py_indicatr = [ None if x == constant.null_price else x for x in indicator]
x = [i-0.2 for i in range(len(indicator))]
y = py_indicatr
axes.bar(x, py_indicatr, width=width, color=color, edgecolor=edgecolor,
*args, **kwargs)
if legend_on:
leg = axes.legend(loc='upper left')
leg.get_frame().set_alpha(0.5)
if text_on:
if not axes.texts:
axes.text(0.01,0.97, label, horizontalalignment='left', verticalalignment='top',
transform=axes.transAxes, color=text_color)
else:
temp_str = axes.texts[0].get_text() + ' ' + label
axes.texts[0].set_text(temp_str)
if zero_on:
ylim = axes.get_ylim()
if ylim[0]<0<ylim[1]:
axes.hlines(0,0,len(indicator))
axes.autoscale_view()
axes.set_xlim(-1, len(indicator)+1)
def main(fileName):
imgName = fileName.replace('csv', 'png')
df = pd.read_csv(fileName, names=['YEAR', 'SEASON', 'BASIN_CD', 'BASIN_MAP', 'STN_CD', 'STN_MAP'])
df['SEASON'].replace(r'^S', '',inplace=True, regex=True)
df['SEASON'].astype(float)
startYear = min(df.YEAR)
endYear = max(df.YEAR)
yearCount = 0
columns = 9
rows = 5
fig, ax_array = plt.subplots(rows, columns, figsize=(55,30))
for i, ax_row in enumerate(ax_array):
for j, axes in enumerate(ax_row):
currentYear = str(startYear + yearCount)
axes.set_title('Year: ' + currentYear)
df2 = df.query('YEAR==' + str(currentYear))
basinCd = df2.BASIN_CD[yearCount * len(df2)]
stnCd = df2.STN_CD[yearCount * len(df2)]
basinLegend = 'BASIN(' + str(basinCd) + ')'
stnLegend = 'STN(' + str(stnCd) + ')'
seasonList = pd.Series([1, 2, 3, 4], dtype='float')
axes.bar(seasonList + 0.14, df2.BASIN_MAP, color = 'b', width = 0.25)
axes.bar(seasonList - 0.14, df2.STN_MAP, color = 'r', width = 0.25)
axes.legend([basinLegend, stnLegend])
axes.set_xlabel('Season', size=12)
axes.set_ylabel('Prcp(mm)', size=12)
axes.set_xticks(range(int(1), int(4) + 1, 1))
axes.set_xticklabels(['Spring', 'Summer', 'Autumn', 'Winter'])
#axes.grid(True)
yearCount = yearCount + 1
savefig('graph_season/' + imgName)
def ibar(
indicator,
new=True,
axes=None,
legend_on=False,
text_on=False,
text_color='k',
label=None,
width=0.4,
color='r',
edgecolor='r',
zero_on=False,
*args,
**kwargs
):
"""绘制indicator柱状图
:param Indicator indicator: Indicator实例
:param axes: 指定的坐标轴
:param new: 是否在新窗口中显示，只在没有指定axes时生效
:param legend_on: 是否打开图例
:param text_on: 是否在左上角显示指标名称及其参数
:param text_color: 指标名称解释文字的颜色，默认为黑色
:param str label: label显示文字信息，text_on 及 legend_on 为 True 时生效
:param zero_on: 是否需要在y=0轴上绘制一条直线
:param width: Bar的宽度
:param color: Bar的颜色
:param edgecolor: Bar边缘颜色
:param args: pylab plot参数
:param kwargs: pylab plot参数
"""
if not indicator:
print("indicator is None")
return
if not axes:
axes = create_figure() if new else gca()
if not label:
label = "%s %.2f" % (indicator.long_name, indicator[-1])
py_indicatr = [None if x == constant.null_price else x for x in indicator]
x = [i - 0.2 for i in range(len(indicator))]
y = py_indicatr
axes.bar(x, py_indicatr, width=width, color=color, edgecolor=edgecolor, *args, **kwargs)
if legend_on:
leg = axes.legend(loc='upper left')
leg.get_frame().set_alpha(0.5)
if text_on:
if not axes.texts:
axes.text(
0.01,
0.97,
label,
horizontalalignment='left',
verticalalignment='top',
transform=axes.transAxes,
color=text_color
)
else:
temp_str = axes.texts[0].get_text() + ' ' + label
axes.texts[0].set_text(temp_str)
if zero_on:
ylim = axes.get_ylim()
if ylim[0] < 0 < ylim[1]:
axes.hlines(0, 0, len(indicator))
axes.autoscale_view()
axes.set_xlim(-1, len(indicator) + 1)
k = indicator.get_context()
if len(k) > 0:
ax_set_locator_formatter(axes, k.get_date_list(), k.get_query().ktype)
def iplot(indicator,
new=True,
axes=None,
legend_on=False,
text_on=False,
text_color='k',
zero_on=False,
label=None,
*args,
**kwargs):
"""绘制indicator曲线
:param Indicator indicator: indicator实例
:param axes: 指定的坐标轴
:param new: 是否在新窗口中显示，只在没有指定axes时生效
:param legend_on: 是否打开图例
:param text_on: 是否在左上角显示指标名称及其参数
:param text_color: 指标名称解释文字的颜色，默认为黑色
:param zero_on: 是否需要在y=0轴上绘制一条直线
:param str label: label显示文字信息，text_on 及 legend_on 为 True 时生效
:param args: pylab plot参数
:param kwargs: pylab plot参数，如：marker（标记类型）、
markerfacecolor（标记颜色）、
markeredgecolor（标记的边缘颜色）
"""
if not indicator:
print("indicator is None")
return
if not axes:
axes = create_figure() if new else gca()
if not label:
label = "%s %.2f" % (indicator.long_name, indicator[-1])
py_indicatr = [None if x == constant.null_price else x for x in indicator]
axes.plot(py_indicatr, '-', label=label, *args, **kwargs)
if legend_on:
leg = axes.legend(loc='upper left')
leg.get_frame().set_alpha(0.5)
if text_on:
if not axes.texts:
axes.text(0.01,
0.97,
label,
horizontalalignment='left',
verticalalignment='top',
transform=axes.transAxes,
color=text_color)
else:
temp_str = axes.texts[0].get_text() + ' ' + label
axes.texts[0].set_text(temp_str)
if zero_on:
ylim = axes.get_ylim()
if ylim[0] < 0 < ylim[1]:
axes.hlines(0, 0, len(indicator))
axes.autoscale_view()
axes.set_xlim(-1, len(indicator) + 1)
def ibar(indicator,
axes=None,
width=0.4,
color='r',
edgecolor='r',
new=True,
legend_on=False,
text_on=False,
text_color='k',
label=None,
zero_on=False,
*args,
**kwargs):
"""绘制indicator曲线
indicator: Indicator实例
axes: 指定的坐标轴
new: 是否在新窗口中显示，只在没有指定axes时生效
legend_on : 是否打开图例
text_on: 是否在左上角显示指标名称及其参数
text_color: 指标名称解释文字的颜色，默认为黑色
zero_on: 是否需要在y=0轴上绘制一条直线
*args, **kwargs : pylab bar参数
"""
if not indicator:
print("indicator is None")
return
if not axes:
axes = create_one_axes_figure() if new else gca()
if not label:
label = "%s %.2f" % (indicator.long_name, indicator[-1])
py_indicatr = [None if x == constant.null_price else x for x in indicator]
x = [i - 0.2 for i in range(len(indicator))]
y = py_indicatr
axes.bar(x,
py_indicatr,
width=width,
color=color,
edgecolor=edgecolor,
*args,
**kwargs)
if legend_on:
leg = axes.legend(loc='upper left')
leg.get_frame().set_alpha(0.5)
if text_on:
if not axes.texts:
axes.text(0.01,
0.97,
label,
horizontalalignment='left',
verticalalignment='top',
transform=axes.transAxes,
color=text_color)
else:
temp_str = axes.texts[0].get_text() + ' ' + label
axes.texts[0].set_text(temp_str)
if zero_on:
ylim = axes.get_ylim()
if ylim[0] < 0 < ylim[1]:
axes.hlines(0, 0, len(indicator))
axes.autoscale_view()
axes.set_xlim(-1, len(indicator) + 1)
