python：将指数ETF基金数据转换为 midi 文件

这是一个将指数ETF基金数据转换为 MIDI音乐的 python程序示例。程序使用 Tushare 获取股票数据，将收盘价映射为 MIDI音符的音高，并且根据交易量控制音量的大小
pip install tushare
python 3.7  tushare==1.2.89
pip install midiutil
  MIDIUtil==1.2.1

编写 stock2mid.py  如下

# -*- coding: utf-8 -*-
"""使用 Tushare 获取股票数据并转换为 MIDI文件需要先安装依赖：pip install tushare midiutil
"""
import sys
import tushare as ts
from midiutil import MIDIFileif len(sys.argv) ==2:code = sys.argv[1]
else:print('usage: python stock2mid.py stockcode ')sys.exit(1)if len(code) !=6:print('stock code length: 6')sys.exit(2)df = ts.get_k_data(code)
if df.empty is True:print(" df is empty ")sys.exit(2)df = df[ df['date'] > '2024-01-01']
if len(df) <60:print(len(df)," < 60")sys.exit(2)
print('len(df)=', len(df))
# DataFrame 重建索引
df = df.reset_index(drop=True)# 提取收盘价序列
prices = df['close'].astype(float).tolist()
# 提取交易量序列
volumes = df['volume'].astype(int).tolist()
max_vol = max(volumes)
# 根据交易量控制音量的大小：将交易量转为音量
vols = [ int(100*i/max_vol)+20 for i in volumes ]
# 使用 zip 函数组合 prices 和 vols 列表
combined = list(zip(prices, vols))
print(combined[0:5])# 数据归一化处理
min_price = min(prices)
max_price = max(prices)
price_range = max_price - min_price
print(f"{code}: max={max_price}, min={min_price}, range={price_range}")
print('max/min:', max_price/min_price)# MIDI参数配置
# C3（MIDI音符范围最小值=48）
min_note = 48
# C5（MIDI音符范围最大值=84）
max_note = 84
if (max_price/min_price) < (84/48):max_note = int(50 * max_price/min_price)
if max_note %2 ==1:max_note += 1
print(f"max_note={max_note}, min_note={min_note} ")
print('max_note/50=', max_note/50)track = 0
channel = 0
tempo = 120    # BPM
volume = 100   # 音量（0-127）# 创建MIDI文件
midi = MIDIFile(1) # 单音轨
midi.addTrackName(track, 0, f"Stock {code}")
midi.addTempo(track, 0, tempo)# 生成音符序列
time_counter = 0  # 时间指针（单位：拍）
for price,vol in combined:if price_range < 0.01:  # 处理价格无波动的情况pitch = (max_note + min_note) // 2else:normalized = (price - min_price) / price_rangepitch = min_note + int(normalized * (max_note - min_note))# 添加音符（每个音符持续1拍，间隔1拍）midi.addNote(track, channel, pitch, time_counter, 1, vol)time_counter += 1# 保存MIDI文件
output_file =f"melo_{code}.mid"
with open(output_file, 'wb') as f:midi.writeFile(f)
print(f" 生成：{output_file}")

以 华泰柏瑞--沪深300ETF (510300)为示例：
运行 python stock2mid.py 510300
生成 melo_510300.mid

编写 play_mid.py 如下

# -*- coding: utf-8 -*-
import os
import sys
import time
from tkinter import filedialog
import traceback
import pygame
from pygame import mixerdef mixer_init():freq = 44100bitsize = -16channels = 2buffer = 2048mixer.init(freq, bitsize, channels, buffer)# optional volume 0 to 1.0mixer.music.set_volume(0.9)def play_mid(file):if mixer.music.get_busy():mixer.music.fadeout(1000)mixer.music.stop()clock = pygame.time.Clock()try:mixer.music.load(file)except:print(traceback.format_exc())mixer.music.play()while mixer.music.get_busy():clock.tick(30)# main()
f1 = ''
if len(sys.argv) ==1:filetypes = [('mid file','.mid'),('mp3 file','.mp3'),('wav file','.wav')]f1 = filedialog.askopenfilename(initialdir='D:/Music', filetypes=filetypes)
elif len(sys.argv) ==2:f1 = sys.argv[1]
else:print('usage: python play_mid.py file1.mid')print('usage: python play_mid.py file1.mp3')print('usage: python play_mid.py file1.wav')sys.exit(1)if not os.path.exists(f1):print(f"{f1} is not exists.")sys.exit(2)fn,ext = os.path.splitext(f1)
if ext.lower() not in ('.mid','.mp3','.wav'):print('.ext is not (.mid , .mp3','.wav')sys.exit(2)print(f1)
mixer_init()
time1 = time.time()
try:play_mid(f1)
except KeyboardInterrupt as ex:# if user hits Ctrl+C then exit# (works only in console mode)mixer.music.fadeout(1000)mixer.music.stop()raise SystemExit from ex
mixer.music.stop()
time2 = time.time()
print("run time: %.3f s" % (time2-time1))

运行 python play_mid.py  melo_510300.mid