create_samplecsv.py 文件源码

def __create_stocklist(self):
        """???????CSV????????????

        Returns:
            ????????????????????????
        """
        def get_row(idx):
            """CSV????1???????????"""
            return ('TEST{}'.format(idx),
                    'Test {}'.format(idx),
                    'Sector {}'.format(idx % 10),)

        header = ('Symbol', 'Name', 'Sector',)
        num_markets = len(const.MARKET_DATA.keys())
        lis = [[
            get_row(idx2) for idx2, row in enumerate(range(self.num_stocks))
            if idx2 % num_markets == idx1
        ] for idx1 in range(len(const.MARKET_DATA.keys()))]

        for idx, (market_id, market_name) in enumerate(const.MARKET_DATA.items()):
            with open(os.path.join(self.dest_dir_stocklist, market_name + '.csv'), 'w',
                      encoding=const.DEFAULT_FILE_ENCODING, newline='') as fp:
                writer = csv.writer(fp, quoting=csv.QUOTE_ALL)
                writer.writerow(header)
                writer.writerows(lis[idx])

        return [row[0] for row in chain(*lis)]  # Symbol??????????