基于鄰域粗糙集算法python實(shí)現(xiàn)

還在跑程序褪那,我這渣筆記本CPU估計(jì)要跑很久,不如來貼個(gè)代碼吧式塌。
這個(gè)是之前一篇粗糙集相對屬性約簡python實(shí)現(xiàn)的一個(gè)小小改進(jìn)
當(dāng)中距離函數(shù)使用的是歐幾里得距離公式,入坑沒多久友浸,代碼寫的不是很好峰尝,請見諒。

# _*_coding:utf-8 _*_
#@Time    :2018/12/22 上午10:45
#@Author  :we2swing
#@FileName: test.py

import pandas as pd
import numpy as np
import math
import matplotlib.pyplot as plt

def basic_set(df):
    basic = {}
    for i in df.drop_duplicates().values.tolist():
        basic[str(i)] = []
        for j, k in enumerate(df.values.tolist()):
            if k == i:
                basic[str(i)].append(j)

    return basic

def Euclidean(x_data):
    n = x_data.shape[1]#代表列數(shù)
    m = x_data.shape[0]#代表行數(shù)
    num = [0] * n
    num2 = np.zeros((m,m))
    #轉(zhuǎn)為list
    arr = np.array(x_data)

    counti,countj,countk,countz = 0,0,0,0
    #遍歷每一行的元素
    for j in arr:
        countk = 0
        for k in arr:
            if countj != countk:
                countz = 0
                # 遍歷每一行的每一個(gè)元素
                for z in k:
                    temp = np.square(z - j[countz])
                    if countz == 0:
                        num[countz] = temp
                    else:
                        #把每一行中每個(gè)元素的差值平方相加
                        num[countz] = num[countz-1] + temp
                    countz = countz + 1
            else:
                num[countz-1] = 0
            num2[countj][countk] = round(math.sqrt(num[countz-1]),1)
            countk = countk + 1
        countj = countj + 1
    return num2

def key_basic(num,k):
    # k = 4  # 設(shè)δ=k
    # k = [4.0,4.1,4.2,4.3,4.4,4.5,4.6,4.7,4.8,4.9,5.0]
    num1 = {}
    counti, countj = 0, 0
    for i in num:
        num1[counti] = []
        countj = 0
        for j in i:
            if j <= k:
                num1[counti].append(countj)
            countj = countj + 1
        counti = counti + 1
    return num1

def compare(data):
    # k = [4.0, 4.1, 4.2, 4.3, 4.4, 4.5, 4.6, 4.7, 4.8, 4.9, 5.0]
    global t1
    t1 = 4.0
    data = data.dropna(axis=0, how='any')
    x_data = data.drop(['judge'], axis=1)
    y_data = data.loc[:, 'judge']
    n = x_data.shape[1]  # 代表列數(shù)
    m = x_data.shape[0]  # 代表行數(shù)

    # 決策屬性基本集
    y_basic_set = sorted([v for k, v in basic_set(y_data).items()])
    num = Euclidean(x_data)
    # print(num)
    x_basic_set = [v for k, v in key_basic(num,t1).items()]

    # γC(D)
    pos = []
    for i in x_basic_set:
        for j in y_basic_set:
            if set(i).issubset(j):
                pos.append(i)
    # pos.sort()#排序
    r_x_y = len(pos) / len(data)  # 也許可以寫一個(gè)card函數(shù)
    print('依賴度r_x_(y):', r_x_y)

    # 計(jì)算每一個(gè)γ(C-a)(D)
    # 獲取條件屬性個(gè)數(shù)為總下標(biāo)存入集合
    # 收集屬性重要性
    imp_attr = []
    columns_num = list(range(len(x_data.columns)))
    for i in columns_num:
        c = columns_num.copy()
        c.remove(i)
        u = data.iloc[:, c]
        num = Euclidean(u)
        u = sorted([v for k, v in key_basic(num,t1).items()])

        # γC(D)
        pos_va = []
        for k in u:
            for j in y_basic_set:
                if set(k).issubset(j):
                    pos_va.append(k)
        r_x_y_2 = len(pos_va) / len(data)
        r_diff = round((r_x_y - r_x_y_2), 4)
        imp_attr.append(r_diff)

    dict_imp = {}
    for o, p in enumerate(imp_attr):
        dict_imp[data.columns[o]] = p

    result = dict_imp
    # print(imp_attr)
    return result

def sepdata(data):
    # 獲取數(shù)據(jù)長度
    len = data.iloc[:, 0].size
    # 將數(shù)據(jù)劃分
    if len % 100 != 0:
        if len > 100:
            num = len // 100 + 1
        else:
            num = 1
    else:
        if len > 100:
            num = int(len / 100)
        else:
            num = 1
    arr = [[]] * num

    count = 0
    for i in arr:
        # 如果數(shù)少于100或者最后一部分?jǐn)?shù)少于100收恢,則放入一個(gè)由數(shù)長決定的數(shù)組
        if num == 1:
            arr[count] = data.iloc[0:len]  # 取100開始武学,取
        elif count == num - 1:
            arr[count] = data.iloc[100 * count:len]
        else:
            arr[count] = data.iloc[100 * count:(count + 1) * 100]
        count = count + 1
    sorted_dict_imp = [[]] * num
    total = [0] * 27
    title = ['C1', 'C2', 'C3', 'C4', 'C5', 'C6', 'C7', 'C8', 'C9', 'C10', 'C11', 'C12', 'C13', 'C14', 'C15', 'C16',
             'C17', 'C18', 'C19', 'C20', 'C21', 'C22', 'C23', 'C24', 'C25', 'C26', 'C27']
    # total = [0] * 16
    # title = ['C1', 'C2', 'C3', 'C4', 'C5', 'C6', 'C7', 'C8', 'C9', 'C10', 'C11', 'C12', 'C13', 'C14', 'C15', 'C16']
    print(sorted_dict_imp)
    count = 0
    for i in arr:
        print('-------------------------------------第%d個(gè)數(shù)據(jù)集數(shù)據(jù)-----------------------------------------' % (count + 1))
        sorted_dict_imp[count] = compare(i)
        count = count + 1
    count1 = 0
    # 將dict的key為C1-Cn的value存入total中保存,并且相加
    for i in sorted_dict_imp:
        count = 0
        if count1 == 0:
            for j in title:
                total[count] = i.get(j)
                count = count + 1
        else:
            for z in title:
                total[count] = i.get(z) + total[count]
                count = count + 1
        count1 = count1 + 1
    # 輸出最終C1-Cn的結(jié)果
    count = 0
    for i in title:
        print(i, ':', round(total[count], 4))
        count = count + 1
    figure(total)

def figure(opt):
    count = 0
    plt.rcParams['font.sans-serif'] = ['SimHei']  # 用來正常顯示中文標(biāo)簽
    title = ['C1', 'C2', 'C3', 'C4', 'C5', 'C6', 'C7', 'C8', 'C9', 'C10', 'C11', 'C12', 'C13', 'C14', 'C15', 'C16',
             'C17', 'C18', 'C19', 'C20', 'C21', 'C22', 'C23', 'C24', 'C25', 'C26', 'C27']
    plt.scatter(title, opt, c='b', s=7)
    plt.plot(title, opt, linewidth=3, alpha=0.7)
    plt.show()

def main():
    data = pd.read_csv(filepath_or_buffer='/Users/we2swing/Desktop/111.csv')
    sepdata(data)

if __name__ == '__main__':
    main()

里面很多代碼延續(xù)的是之前上面?zhèn)魉烷T里文章的代碼。
覺得有用的話點(diǎn)個(gè)贊
測試數(shù)據(jù):
鏈接:https://pan.baidu.com/s/1dtle6A6od8bOCwDPW1lZ2w
提取碼:o2qf

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末伦意,一起剝皮案震驚了整個(gè)濱河市火窒,隨后出現(xiàn)的幾起案子,更是在濱河造成了極大的恐慌驮肉,老刑警劉巖熏矿,帶你破解...
    沈念sama閱讀 206,968評論 6 482
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件,死亡現(xiàn)場離奇詭異离钝,居然都是意外死亡票编,警方通過查閱死者的電腦和手機(jī),發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 88,601評論 2 382
  • 文/潘曉璐 我一進(jìn)店門卵渴,熙熙樓的掌柜王于貴愁眉苦臉地迎上來慧域,“玉大人,你說我怎么就攤上這事浪读∥袅瘢” “怎么了?”我有些...
    開封第一講書人閱讀 153,220評論 0 344
  • 文/不壞的土叔 我叫張陵碘橘,是天一觀的道長互订。 經(jīng)常有香客問我,道長蛹屿,這世上最難降的妖魔是什么屁奏? 我笑而不...
    開封第一講書人閱讀 55,416評論 1 279
  • 正文 為了忘掉前任,我火速辦了婚禮错负,結(jié)果婚禮上坟瓢,老公的妹妹穿的比我還像新娘勇边。我一直安慰自己,他們只是感情好折联,可當(dāng)我...
    茶點(diǎn)故事閱讀 64,425評論 5 374
  • 文/花漫 我一把揭開白布粒褒。 她就那樣靜靜地躺著,像睡著了一般诚镰。 火紅的嫁衣襯著肌膚如雪奕坟。 梳的紋絲不亂的頭發(fā)上,一...
    開封第一講書人閱讀 49,144評論 1 285
  • 那天清笨,我揣著相機(jī)與錄音月杉,去河邊找鬼。 笑死抠艾,一個(gè)胖子當(dāng)著我的面吹牛苛萎,可吹牛的內(nèi)容都是我干的。 我是一名探鬼主播检号,決...
    沈念sama閱讀 38,432評論 3 401
  • 文/蒼蘭香墨 我猛地睜開眼腌歉,長吁一口氣:“原來是場噩夢啊……” “哼!你這毒婦竟也來了齐苛?” 一聲冷哼從身側(cè)響起翘盖,我...
    開封第一講書人閱讀 37,088評論 0 261
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎凹蜂,沒想到半個(gè)月后馍驯,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體,經(jīng)...
    沈念sama閱讀 43,586評論 1 300
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡炊甲,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 36,028評論 2 325
  • 正文 我和宋清朗相戀三年泥彤,在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片卿啡。...
    茶點(diǎn)故事閱讀 38,137評論 1 334
  • 序言:一個(gè)原本活蹦亂跳的男人離奇死亡吟吝,死狀恐怖,靈堂內(nèi)的尸體忽然破棺而出颈娜,到底是詐尸還是另有隱情剑逃,我是刑警寧澤,帶...
    沈念sama閱讀 33,783評論 4 324
  • 正文 年R本政府宣布官辽,位于F島的核電站蛹磺,受9級特大地震影響,放射性物質(zhì)發(fā)生泄漏同仆。R本人自食惡果不足惜萤捆,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 39,343評論 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧俗或,春花似錦市怎、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 30,333評論 0 19
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至帅腌,卻和暖如春驰弄,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背速客。 一陣腳步聲響...
    開封第一講書人閱讀 31,559評論 1 262
  • 我被黑心中介騙來泰國打工戚篙, 沒想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留,地道東北人溺职。 一個(gè)月前我還...
    沈念sama閱讀 45,595評論 2 355
  • 正文 我出身青樓已球,卻偏偏與公主長得像,于是被迫代替她去往敵國和親辅愿。 傳聞我的和親對象是個(gè)殘疾皇子,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 42,901評論 2 345