mdj1412 commited on
Commit
55c7203
·
1 Parent(s): 3a66a23
Files changed (2) hide show
  1. app.py +0 -1
  2. dataset_creation/nasdaq_data.py +8 -3
app.py CHANGED
@@ -1,5 +1,4 @@
1
  import os
2
- from bs4 import BeautifulSoup
3
  import pandas as pd
4
 
5
  from flask import Flask, jsonify, request, render_template
 
1
  import os
 
2
  import pandas as pd
3
 
4
  from flask import Flask, jsonify, request, render_template
dataset_creation/nasdaq_data.py CHANGED
@@ -27,11 +27,16 @@ def get_list(tickers=nasdaq100_symbols):
27
  ticker = demo_dic.loc[i, 'ticker']
28
 
29
  # ticker의 주식 정보 데이터를 가져온다.
30
- data = get_data(tickers=[ticker], numOfDay=3)[0]
 
 
 
 
 
 
 
31
 
32
 
33
- yesterday = data.iloc[-2, 3]
34
- today = data.iloc[-1, 3]
35
  demo_dic.loc[i, 'diff'] = round(((today-yesterday)/today) * 100.0, 2)
36
  demo_dic.loc[i, 'open'] = round(data.iloc[-1, 0], 2) # Open
37
  demo_dic.loc[i, 'close'] = round(data.iloc[-1, 3], 2) # Close
 
27
  ticker = demo_dic.loc[i, 'ticker']
28
 
29
  # ticker의 주식 정보 데이터를 가져온다.
30
+ data = get_data(tickers=[ticker], numOfDay=4)[0]
31
+
32
+ try:
33
+ yesterday = data.iloc[-2, 3]
34
+ today = data.iloc[-1, 3]
35
+ except IndexError:
36
+ print("Oops! That was no valid number. Try again...")
37
+ from IPython import embed; embed()
38
 
39
 
 
 
40
  demo_dic.loc[i, 'diff'] = round(((today-yesterday)/today) * 100.0, 2)
41
  demo_dic.loc[i, 'open'] = round(data.iloc[-1, 0], 2) # Open
42
  demo_dic.loc[i, 'close'] = round(data.iloc[-1, 3], 2) # Close