-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathreadcsv.py
More file actions
33 lines (26 loc) · 927 Bytes
/
readcsv.py
File metadata and controls
33 lines (26 loc) · 927 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
# -*- coding: UTF-8 -*-
__author__ = 'chun'
import csv
def readcsv():
twse_id_list = []
gtsm_id_list = []
stock_id_dict = {}
csv_file_list = ['TWSE_price_ranking.csv', 'GTSM_price_ranking.csv']
for filename in csv_file_list:
with open(filename, 'rb') as csvfile:
twsereader = csv.DictReader(csvfile, ['代碼', '名稱', '成交'])
for row in twsereader:
if row['代碼'] == '代碼':
pass
else:
if filename == csv_file_list[0]:
twse_id_list.append(row['代碼'])
if filename == csv_file_list[1]:
gtsm_id_list.append(row['代碼'])
csvfile.close()
stock_id_dict['TWSE_ID_LIST'] = twse_id_list
stock_id_dict['GTSM_ID_LIST'] = gtsm_id_list
# print stock_id_dict
return stock_id_dict
# debug log
# readcsv()