零基础用Python抓取股票数据,手把手教你入门量化投资!

pip install tushare pandas matplotlib

import tushare as tsimport pandas as pdimport matplotlib.pyplot as plt# 初始化tushare（需要注册获取token）# 前往 tushare.pro 注册免费获取ts.set_token('你的token')  # 替换为你的实际tokenpro = ts.pro_api()# 获取贵州茅台(600519)日线数据df = pro.daily(ts_code='600519.SH',                start_date='20230101',                end_date='20231231')print(f"数据形状：{df.shape}")print(df.head())

# 转换日期格式df['trade_date'] = pd.to_datetime(df['trade_date'])# 绘制股价走势图plt.figure(figsize=(12, 6))plt.plot(df['trade_date'], df['close'], label='收盘价', linewidth=2)plt.title('贵州茅台2023年股价走势', fontsize=15)plt.xlabel('日期', fontsize=12)plt.ylabel('价格(元)', fontsize=12)plt.legend()plt.grid(True, alpha=0.3)plt.xticks(rotation=45)plt.tight_layout()plt.show()

# 获取实时行情realtime_df = ts.get_realtime_quotes('600519')  # 贵州茅台# 只保留关键字段important_columns = ['name', 'open', 'pre_close', 'price', 'high', 'low', 'volume']realtime_info = realtime_df[important_columns]print("实时行情：")print(realtime_info)

# 计算简单技术指标df['MA5'] = df['close'].rolling(5).mean()  # 5日均线df['MA20'] = df['close'].rolling(20).mean()  # 20日均线# 计算日收益率df['daily_return'] = df['close'].pct_change()print("描述性统计：")print(df[['close', 'daily_return']].describe())