zwq2018 commited on
Commit
9bac4cf
1 Parent(s): a14618a

Upload tool.py

Browse files
Files changed (1) hide show
  1. tool.py +5 -5
tool.py CHANGED
@@ -384,22 +384,22 @@ def plot_stock_data(stock_data: pd.DataFrame, ax: Optional[plt.Axes] = None, fig
384
 
385
  ax.annotate(unchanged_var + ':' + str(round(start_y, 2)) + ' @' + start_x_dim, xy=(start_x_dim, start_y),
386
  xytext=(start_x_dim, start_y),
387
- textcoords='data', fontsize=14,color=color, horizontalalignment='right')
388
 
389
  ax.annotate(unchanged_var + ':' + str(round(end_y, 2)) +' @' + end_x_dim, xy=(end_x_dim, end_y),
390
  xytext=(end_x_dim, end_y),
391
- textcoords='data', fontsize=14, color=color, horizontalalignment='left')
392
 
393
 
394
  elif figure_type == 'bar':
395
  ax.bar(x_dim, stock_data.iloc[:, 2], label = unchanged_var + '_' + index_name, width=0.3, color=color)
396
  ax.annotate(unchanged_var + ':' + str(round(start_y, 2)) + ' @' + start_x_dim, xy=(start_x_dim, start_y),
397
  xytext=(start_x_dim, start_y),
398
- textcoords='data', fontsize=14, color=color, horizontalalignment='right')
399
 
400
  ax.annotate(unchanged_var + ':' + str(round(end_y, 2)) + ' @' + end_x_dim, xy=(end_x_dim, end_y),
401
  xytext=(end_x_dim, end_y),
402
- textcoords='data', fontsize=14, color=color, horizontalalignment='left')
403
 
404
  plt.xticks(x_dim,rotation=45) #
405
  ax.xaxis.set_major_locator(MaxNLocator( integer=True, prune=None, nbins=100)) #
@@ -409,7 +409,7 @@ def plot_stock_data(stock_data: pd.DataFrame, ax: Optional[plt.Axes] = None, fig
409
  plt.ylabel(f'{index_name}', fontproperties=font_prop,fontsize=16)
410
  ax.set_title(title_name , fontproperties=font_prop,fontsize=16)
411
  plt.legend() # 显示图例
412
- fig = plt.gcf()
413
  fig.set_size_inches(18, 12)
414
 
415
  return ax
 
384
 
385
  ax.annotate(unchanged_var + ':' + str(round(start_y, 2)) + ' @' + start_x_dim, xy=(start_x_dim, start_y),
386
  xytext=(start_x_dim, start_y),
387
+ textcoords='data', fontsize=14,color=color, horizontalalignment='right',fontproperties=font_prop)
388
 
389
  ax.annotate(unchanged_var + ':' + str(round(end_y, 2)) +' @' + end_x_dim, xy=(end_x_dim, end_y),
390
  xytext=(end_x_dim, end_y),
391
+ textcoords='data', fontsize=14, color=color, horizontalalignment='left',fontproperties=font_prop)
392
 
393
 
394
  elif figure_type == 'bar':
395
  ax.bar(x_dim, stock_data.iloc[:, 2], label = unchanged_var + '_' + index_name, width=0.3, color=color)
396
  ax.annotate(unchanged_var + ':' + str(round(start_y, 2)) + ' @' + start_x_dim, xy=(start_x_dim, start_y),
397
  xytext=(start_x_dim, start_y),
398
+ textcoords='data', fontsize=14, color=color, horizontalalignment='right',fontproperties=font_prop)
399
 
400
  ax.annotate(unchanged_var + ':' + str(round(end_y, 2)) + ' @' + end_x_dim, xy=(end_x_dim, end_y),
401
  xytext=(end_x_dim, end_y),
402
+ textcoords='data', fontsize=14, color=color, horizontalalignment='left',fontproperties=font_prop)
403
 
404
  plt.xticks(x_dim,rotation=45) #
405
  ax.xaxis.set_major_locator(MaxNLocator( integer=True, prune=None, nbins=100)) #
 
409
  plt.ylabel(f'{index_name}', fontproperties=font_prop,fontsize=16)
410
  ax.set_title(title_name , fontproperties=font_prop,fontsize=16)
411
  plt.legend() # 显示图例
412
+ fig = plt.gcf(prop=font_properties)
413
  fig.set_size_inches(18, 12)
414
 
415
  return ax