Python Pandas 读取 Excel 文件并提取上市公司信息
def get_company_info(file_name):
'输入参数：file_name，字符串类型，表示Excel文件名
输出：元组类型，包含两个列表，第一个列表为上市公司的名字列表，第二个列表为股票代码列表
'
wb = openpyxl.load_workbook(file_name, read_only=True)
name_lst = wb.sheetnames
stock_name = name_lst[0]
df_stock_values = pd.read_excel(file_name, sheet_name=stock_name)
lst_name = list(df_stock_values['ts_code'].apply(lambda x: x.split('.')[0]))
lst_code = list(df_stock_values['ts_code'].apply(lambda x: x.split('.')[1]))
return (lst_name, lst_code)
原文地址: https://www.cveoy.top/t/topic/nj2P 著作权归作者所有。请勿转载和采集!