TOPSIS综合评价法及其Python实现(2)

TOPSIS综合评价法及其Python实现

式中M为:

完成正向化后，即可进行标准化或归一化。

代码

以下为 TOPSIS 算法的代码，使用了两种权重添加方法。

import numpy as np
import pandas as pd
from sklearn.datasets import make_regression
from sklearn.preprocessing import StandardScaler
def topsis(data, weights=0, use_weights=0):
    '''
    use_weights == 0 表明不使用权重
    use_weights == 1 表明使用数据权重
    use_weights == 2 表明使用方法权重
    '''
    data = np.array(data)
    data = StandardScaler().fit_transform(data)
    m, n = data.shape
    if use_weights == 1: # # ! 加权方法 1
        data = data * weights  # 计算加权标准化矩阵
    dataMax = data.max(axis=0) # 最优解
    dataMin = data.min(axis=0) # 最劣解
    D_max_ij = np.zeros([m,n])
    D_min_ij = np.zeros([m,n])
    # 计算距离 
    if use_weights == 2: # ! 加权方法 2
        D_max_ij = ((dataMax - data) ** 2) * weights
        D_min_ij = ((dataMin - data) ** 2) * weights
    else:
        D_max_ij = (dataMax - data) ** 2
        D_min_ij = (dataMin - data) ** 2
            
    D_max_i = np.sqrt(D_max_ij.sum(axis=1))
    D_min_i = np.sqrt(D_min_ij.sum(axis=1))
    S_i = D_min_i / (D_min_i + D_max_i)
    return pd.DataFrame(S_i)
    
X, y = make_regression(n_samples=200, n_features=4, noise=0.5, random_state=10)
topsis(X, weights=(0.1, 0.5, 0.2, 0.2), use_weights=2)

以上分享的TOPSIS综合评价法及其Python实现的具体内容，未经IT袋网网（www.itdai.com）允许禁止转载！