add code

Author: itdoudoutech

doudou/2020-03-27-found/found_two.py

@@ -1,28 +1,24 @@
 import json
 import datetime
 import calendar
-import matplotlib.pyplot as plt
-from matplotlib import font_manager
-import numpy as np
+from pyecharts.charts import Bar
+from pyecharts.charts import Line
+from pyecharts import options as opts
 
 foundCode = '510300'
-file = f'./found_{foundCode}.txt'
-found_date_price = {}
-found_price_x = []
-found_price_y = []
+fixed_investment_amount_per_week = 500 # 每周定投金额
+fixed_investment_amount_per_month = 2000 # 每月定投金额
 
-fixed_investment_amount_per_week = 500
-fixed_investment_amount_per_month = 2000
-my_font = font_manager.FontProperties(fname='/System/Library/Fonts/PingFang.ttc')
-
-with open(file) as f:
-    line = f.readline()
-    result = json.loads(line)
-    for found in result['Data']['LSJZList'][::-1]:
-        found_date_price[found['FSRQ']] = found['DWJZ']
-        found_price_x.append(found['FSRQ'])
-        found_price_y.append(found['DWJZ'])
+def get_data():
+    with open(f'./found_{foundCode}.txt') as f:
+        line = f.readline()
+        result = json.loads(line)
+        found_date_price = {}
+        for found in result['Data']['LSJZList'][::-1]:
+            found_date_price[found['FSRQ']] = found['DWJZ']
+        return found_date_price
 
+found_date_price = get_data()
 
 # 买入规则：从 start_date 日期开始，每逢 weekday 买入，如果 weekday 不是交易日，则顺延至最近的交易日
 # 每次买入 500 元，之后转化为相应的份额
@@ -42,17 +38,15 @@ def calculate_found_profit_by_week(start_date, end_date, weekday):
         nums += 1
         total_stock += round(fixed_investment_amount_per_week / float(found_date_price[day.strftime('%Y-%m-%d')]), 2)
         total_amount += fixed_investment_amount_per_week
-        # print(day.strftime('%Y-%m-%d'), found_date_price[day.strftime('%Y-%m-%d')],
-        # round(fixed_investment_amount / float(found_date_price[day.strftime('%Y-%m-%d')]), 2), nums)
 
     # 计算盈利
     while found_date_price.get(end_date.strftime('%Y-%m-%d'), None) is None:
         end_date += datetime.timedelta(days=-1)
+    
     total_profit = round(total_stock, 2) * float(found_date_price[end_date.strftime('%Y-%m-%d')]) - total_amount
 
     return nums, round(total_stock, 2), total_amount, round(total_profit)
 
-
 def get_first_day_of_next_month(date):
     first_day = datetime.datetime(date.year, date.month, 1)
     days_num = calendar.monthrange(first_day.year, first_day.month)[1]  # 获取一个月有多少天
@@ -76,8 +70,6 @@ def calculate_found_profit_by_month(start_date, end_date):
         nums += 1
         total_stock += round(fixed_investment_amount_per_month / float(found_date_price[day.strftime('%Y-%m-%d')]), 2)
         total_amount += fixed_investment_amount_per_month
-        # print(day.strftime('%Y-%m-%d'), found_date_price[day.strftime('%Y-%m-%d')],
-        # round(fixed_investment_amount / float(found_date_price[day.strftime('%Y-%m-%d')]), 2), nums)
 
     # 计算盈利
     while found_date_price.get(end_date.strftime('%Y-%m-%d'), None) is None:
@@ -86,87 +78,54 @@ def calculate_found_profit_by_month(start_date, end_date):
 
     return nums, round(total_stock, 2), total_amount, round(total_profit)
 
-
 start_date = datetime.datetime.fromisoformat('2010-01-01')
 end_date = datetime.datetime.fromisoformat('2020-03-01')
 
-
 def calculate_found_profit_week_month():
     total_amount = []
     total_profit = []
-
+    # 周定投收益
     for i in range(5):
         result = calculate_found_profit_by_week(start_date, end_date, i)
         total_amount.append(result[2])
         total_profit.append(result[3])
-
+    # 月定投收益
     result_month = calculate_found_profit_by_month(start_date, end_date)
     total_amount.append(result_month[2])
     total_profit.append(result_month[3])
     return total_amount, total_profit
 
-
 total_amount, total_profit = calculate_found_profit_week_month()
 
-print(total_amount)
-print(total_profit)
-
-fig, ax = plt.subplots()
-
-
-## 柱状图
-def auto_text(rects):
-    for rect in rects:
-        ax.text(rect.get_x(), rect.get_height(), rect.get_height(), ha='left', va='bottom')
-
-
-def show_pic():
-    labels = ['周一', '周二', '周三', '周四', '周五', '月定投']
-    index = np.arange(len(labels))
-    width = 0.2
-
-    fig, ax = plt.subplots()
-    rect1 = ax.bar(index - width / 2, total_profit, color='lightcoral', width=width, label='投资收益')
-    rect2 = ax.bar(index + width / 2, total_amount, color='springgreen', width=width, label='投资金额')
-
-    plt.title("投入金额 & 收益柱状图", fontproperties=my_font)
-    plt.xticks(fontproperties=my_font)
-    ax.set_xticks(ticks=index)
-    ax.set_xticklabels(labels)
-
-    ax.set_ylim(0, 220000)
-    auto_text(rect1)
-    auto_text(rect2)
-
-    plt.show()
-
-
-# show_pic()
-
-
-# 基金走势
-def show_found(found_price_y):
-    found_price_y = list(map(float, found_price_y))
-    x = [i for i in range(0, len(found_price_y))]
-
-    plt.figure(figsize=(10, 6))
-
-    plt.plot(x, found_price_y, linewidth=1, color='r')
-
-    plt.xlabel('时间', fontproperties=my_font)
-    plt.ylabel('单位净值', fontproperties=my_font)
-    plt.title(f"{foundCode} 基金走势", fontproperties=my_font)
-    plt.xticks(x[::90], found_price_x[::90], rotation=45)
-
-    plt.show()
-
-
-# show_found(found_price_y)
-
-def calculate_found_profit():
-    start_date = datetime.datetime.fromisoformat('2015-06-10')
-    end_date = datetime.datetime.fromisoformat('2020-03-01')
-    result = calculate_found_profit_by_month(start_date, end_date)
-    print(result)
-
-# calculate_found_profit()
+# 这部分代码在 jupyter 中 run
+line = (
+    Line()
+    .add_xaxis(list(found_date_price.keys()))
+    .add_yaxis('price',list(found_date_price.values()),label_opts=opts.LabelOpts(is_show=False))
+    .set_global_opts(
+        title_opts=opts.TitleOpts(title=f'{foundCode}基金走势图'),
+        xaxis_opts=opts.AxisOpts(splitline_opts=opts.SplitLineOpts(is_show=True)),
+        yaxis_opts=opts.AxisOpts(splitline_opts=opts.SplitLineOpts(is_show=True)),
+    )
+)
+#line.render_notebook()
+
+x = ['周一', '周二', '周三', '周四', '周五', '月定投']
+bar = (
+    Bar()
+    .add_xaxis(x)
+    .add_yaxis('投资金额', total_amount)
+    .add_yaxis('投资收益', total_profit)
+    .set_global_opts(
+        title_opts=opts.TitleOpts(title="投资总额 & 投资收益"),
+        xaxis_opts=opts.AxisOpts(splitline_opts=opts.SplitLineOpts(is_show=True)),
+        yaxis_opts=opts.AxisOpts(splitline_opts=opts.SplitLineOpts(is_show=True)),
+    )
+)
+#bar.render_notebook()
+# 这部分代码在 jupyter 中 run
+
+start_date = datetime.datetime.fromisoformat('2015-06-10')
+end_date = datetime.datetime.fromisoformat('2020-03-01')
+result = calculate_found_profit_by_month(start_date, end_date)
+print(result)

doudou/2020-04-04-greedy-snake/db.txt

@@ -0,0 +1 @@
+5

doudou/2020-04-04-greedy-snake/greedy_snake.py

@@ -0,0 +1,183 @@
+import pygame, sys, random
+from pygame.locals import *
+
+# 颜色配置
+snake_color = pygame.Color("#8B7D1C")
+food_color = pygame.Color("#8B0000")
+background_color = pygame.Color("#BACB03")
+text_color = pygame.Color("#EFEFEF")
+
+speed = 5
+# 长度单位
+pixel = 15
+line = 44
+row = 36
+window_width = pixel * line
+window_high = pixel * row
+
+point_left_up = [pixel * 2, pixel * 4]
+point_left_down = [pixel * 2, pixel * (row - 2)]
+point_right_up = [pixel * (line - 2), pixel * 4]
+point_right_down = [pixel * (line - 2), pixel * (row - 2)]
+
+# 蛇头位置
+snake_head = [pixel * 8, pixel * 8]
+# 蛇身
+snake_body = [[snake_head[0] - x * pixel, snake_head[1]] for x in range(5)]
+# 方向
+direction_right = 0
+direction_up = 90
+direction_left = 180
+direction_down = 270
+move = {direction_right: [pixel, 0], direction_left: [-pixel, 0],
+        direction_up: [0, -pixel], direction_down: [0, pixel]}
+
+# 分数设置
+score = 5
+filename = 'db.txt'
+
+
+def write_score(content):
+    with open(filename, 'w+') as f:
+        f.write(str(content))
+
+
+def read_score():
+    with open(filename, 'w+') as f:
+        result = f.readline()
+        return 0 if result.strip() == '' else int(result)
+
+
+def init():
+    # 初始化
+    pygame.init()
+    # new a window
+    my_screen = pygame.display.set_mode((window_width, window_high), 0, 32)
+    # 设置标题
+    pygame.display.set_caption("Greedy Snake")
+    return my_screen
+
+
+# 游戏结束
+def game_over(max_score, current_score):
+    if max_score < current_score:
+        write_score(current_score)
+    pygame.quit()
+    sys.exit()
+
+
+screen = init()
+time_clock = pygame.time.Clock()
+
+
+# 画边线
+def draw_box():
+    for point in [[point_left_up, point_right_up], [point_right_up, point_right_down],
+                  [point_right_down, point_left_down], [point_left_down, point_left_up]]:
+        pygame.draw.line(screen, snake_color, point[0], point[1], 1)
+
+
+def is_alive():
+    # 越界 -> game over
+    if (snake_head[0] < point_left_up[0] or snake_head[0] > (point_right_down[0] - pixel) or
+            snake_head[1] < point_left_up[1] or snake_head[1] > (point_right_down[1]) - pixel):
+        return False
+
+    # 头触碰到身体 -> game over
+    if snake_head in snake_body:
+        return False
+
+    return True
+
+
+# 随机产生食物
+def create_food():
+    while True:
+        x = random.randint(point_left_up[0] / pixel, point_right_down[0] / pixel - 1) * pixel
+        y = random.randint(point_left_up[1] / pixel, point_right_down[1] / pixel - 1) * pixel
+        if [x, y] not in snake_body:
+            break
+    return [x, y]
+
+
+def draw_snake(food_position):
+    # 画蛇
+    for point in snake_body:
+        pygame.draw.rect(screen, snake_color, Rect(point[0], point[1], pixel, pixel))
+    # 画食物
+    pygame.draw.rect(screen, food_color, Rect(food_position[0], food_position[1], pixel, pixel))
+
+
+def display_message(text, color, size, postion):
+    font = pygame.font.Font(None, size)
+    text = font.render(text, True, color)
+    screen.blit(text, postion)
+    pygame.display.update()
+
+
+# 入口函数
+def run():
+    food_position = create_food()
+    max_score = read_score()
+    current_score = 0
+    is_dead = False
+    origin_direction = direction_right
+    target_direction = origin_direction
+    while True:
+        # 监听键盘按键 退出 OR 换方向
+        for event in pygame.event.get():
+            if event.type == pygame.QUIT:
+                game_over(max_score, current_score)
+            if event.type == KEYDOWN:
+                # 方向键 or asdw 控制方向
+                if event.key == K_RIGHT or event.key == K_d:
+                    target_direction = direction_right
+                if event.key == K_LEFT or event.key == K_a:
+                    target_direction = direction_left
+                if event.key == K_UP or event.key == K_w:
+                    target_direction = direction_up
+                if event.key == K_DOWN or event.key == K_s:
+                    target_direction = direction_down
+                # esc 退出
+                if event.key == K_ESCAPE:
+                    game_over(max_score, current_score)
+            # 夹角为 90 or 270 可以转换方向
+            angle = abs(origin_direction - target_direction)
+            if angle == 90 or angle == 270:
+                origin_direction = target_direction
+
+        if not is_dead:
+            snake_head[0] += move[origin_direction][0]
+            snake_head[1] += move[origin_direction][1]
+
+        if not is_dead and is_alive():
+            # 按 origin_direction 方向运动
+            snake_body.insert(0, list(snake_head))
+            # 吃到食物后重新生成
+            if snake_head[0] == food_position[0] and snake_head[1] == food_position[1]:
+                food_position = create_food()
+                current_score += score
+            else:
+                # 移除最后一格
+                snake_body.pop()
+        else:
+            is_dead = True
+
+        # 画背景
+        screen.fill(background_color)
+        # 画边框
+        draw_box()
+        # 画蛇
+        draw_snake(food_position)
+        # 刷新画面
+        pygame.display.update()
+        # 更新分数
+        display_message(f"{current_score}/{max_score}", text_color, 30, (pixel * 2, pixel * 2))
+        if is_dead:
+            display_message(f"Game Over", text_color, 50, (pixel * 16, pixel * 15))
+        # 控制游戏速度
+        time_clock.tick(speed)
+
+
+if __name__ == '__main__':
+    run()