liuyu/qlib
1
0
Fork 0
mirror of https://github.com/microsoft/qlib.git synced 2026-07-18 18:04:31 +08:00
Code Activity

black

Browse Source
This commit is contained in:
wangershi
2021-03-07 19:35:50 +08:00
parent 6bcd88973b
commit 9df0361262
1 changed files with 13 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
scripts/data_collector/utils.py
View File

@@ -105,7 +105,11 @@ def return_date_list(source_dir, date_field_name: str, file_path: Path):
def get_calendar_list_by_ratio(
source_dir: [str, Path], date_field_name: str = "date", threshold: float = 0.5, minimum_count: int = 10, max_workers: int = 16
source_dir: [str, Path],
date_field_name: str = "date",
threshold: float = 0.5,
minimum_count: int = 10,
max_workers: int = 16,
) -> list:
"""get calendar list by selecting the date when few funds trade in this day
@@ -157,7 +161,11 @@ def get_calendar_list_by_ratio(
if date < oldest_date:
_dict_count_founding[date] -= 1
calendar = [date for date in _dict_count_trade if _dict_count_trade[date] >= max(int(_dict_count_founding[date] * threshold), minimum_count)]
calendar = [
date
for date in _dict_count_trade
if _dict_count_trade[date] >= max(int(_dict_count_founding[date] * threshold), minimum_count)
]
return calendar