用Python程序写出求DMI指标

好的，我可以为您编写一个Python函数来计算DMI指标。DMI指标是一种技术分析指标，用于评估趋势强度和方向。

首先，我们需要计算出来True Range(TR)、+DI和-DI指标，然后再根据公式计算出DMI指标。

以下是Python代码，可以帮助您计算DMI指标：

import pandas as pd

def calculate_dmi(data, period=14):
    # 计算True Range(TR)
    data['h_l'] = abs(data['High'] - data['Low'])
    data['h_pc'] = abs(data['High'] - data['Close'].shift(1))
    data['l_pc'] = abs(data['Low'] - data['Close'].shift(1))
    data['TR'] = data[['h_l', 'h_pc', 'l_pc']].max(axis=1)
    
    # 计算+DI和-DI指标
    data['+DM'] = (data['High'] - data['High'].shift(1)).apply(lambda x: x if x > 0 else 0)
    data['-DM'] = (data['Low'].shift(1) - data['Low']).apply(lambda x: x if x > 0 else 0)
    data['TR_14'] = data['TR'].rolling(window=period).sum()
    data['+DM_14'] = data['+DM'].rolling(window=period).sum()
    data['-DM_14'] = data['-DM'].rolling(window=period).sum()
    data['+DI_14'] = (data['+DM_14'] / data['TR_14']) * 100
    data['-DI_14'] = (data['-DM_14'] / data['TR_14']) * 100
    
    # 计算ADX指标
    data['DX'] = ((data['+DI_14'] - data['-DI_14']).abs() / (data['+DI_14'] + data['-DI_14'])) * 100
    data['ADX'] = data['DX'].rolling(window=period).mean()
    
    # 计算DMI指标
    data['DMI'] = (data['ADX'] / 100) * data['DX']
    
    return data[['Date', 'DMI']]

这个函数需要一个包含日期、最高价、最低价和收盘价的数据框(dataframe)作为输入，同时还可以指定计算DMI指标所需要的时间窗口(period)。

函数返回一个数据框，其中包含日期和DMI指标。

希望这个函数能帮助您计算DMI指标。如果您有任何疑问，请随时问我！