30 lines
918 B
Python
30 lines
918 B
Python
import os
|
||
import tushare as ts
|
||
|
||
# 从环境变量读取Token(需提前设置环境变量TUSHARE_TOKEN)
|
||
ts.set_token('9343e641869058684afeadfcfe7fd6684160852e52e85332a7734c8d')
|
||
|
||
try:
|
||
# 定义股票代码列表,提升可维护性
|
||
# sina数据
|
||
|
||
stock_codes = ['600000.SH', '000001.SZ', '000001.SH']
|
||
df = ts.realtime_quote(ts_code=','.join(stock_codes))
|
||
print(df)
|
||
|
||
# 检查数据有效性
|
||
if not df.empty:
|
||
print("所有列名:", df.columns.tolist())
|
||
|
||
# 根据实际列名调整(假设列名为小写)
|
||
required_columns = ['HIGH', 'LOW', 'PRICE']
|
||
if all(col in df.columns for col in required_columns):
|
||
print(df[required_columns])
|
||
else:
|
||
print("列名不匹配,请检查数据源列名格式")
|
||
else:
|
||
print("获取数据失败,返回空DataFrame")
|
||
except Exception as e:
|
||
print(f"发生错误: {str(e)}")
|
||
|