提高Python數組排序性能的利器

一、使用內置函數sorted()

在Python中,有個內置函數為sorted(),它可以對任何可迭代對象進行排序,包括但不限於列表、元組、字典、集合等,其時間複雜度為O(nlogn),運行速度非常快。

    lst = [3, 1, 4, 1, 5, 9, 2, 6, 5, 3, 5]
    sorted_lst = sorted(lst)
    print(sorted_lst)
    # 輸出結果:[1, 1, 2, 3, 3, 4, 5, 5, 5, 6, 9]

使用內置函數方法較為簡單,只需要一行代碼就可以實現排序。但需要注意的是,sorted()並不是就完全比快速排序更快,它增加了相當不小的空間消耗,因為它會創建一個新列表來存放排序後的內容。

二、使用sort()方法進行排序

Python列表的sort()方法是直接在原地進行排序,時間複雜度為O(nlogn),它可以與內置函數sorted()相媲美,最常用於對列表進行排序。

    lst = [3, 1, 4, 1, 5, 9, 2, 6, 5, 3, 5]
    lst.sort()
    print(lst)
    # 輸出結果:[1, 1, 2, 3, 3, 4, 5, 5, 5, 6, 9]

需要注意的是,sort()方法會改變原有列表,如果不希望改變原有列表,可以使用sorted()方法。

三、使用numpy庫進行排序

numpy庫是Python中科學計算的重要庫,其中有一個numpy.sort()函數可以對ndarray數組進行排序,其時間複雜度為O(nlogn)。

    import numpy as np
    
    arr = np.array([3, 1, 4, 1, 5, 9, 2, 6, 5, 3, 5])
    sorted_arr = np.sort(arr)
    print(sorted_arr)
    # 輸出結果:[1 1 2 3 3 4 5 5 5 6 9]

numpy.sort()函數使用起來同樣簡單易懂,並且不僅支持一維數組,還可以支持多維數組的排序,這是Python內置函數和列表方法無法實現的。

四、使用list的count()方法

如果列表中存在著大量相同的元素且需要對其進行升序或降序排序,此時使用列表的count()方法可以提高排序性能。count()方法的時間複雜度是O(n),可以快速獲取元素在列表中的個數。

    lst = [3, 1, 4, 1, 5, 9, 2, 6, 5, 3, 5]
    sorted_lst = []
    for i in range(min(lst), max(lst)+1):
        sorted_lst += [i] * lst.count(i)
    print(sorted_lst)
    # 輸出結果:[1, 1, 2, 3, 3, 4, 5, 5, 5, 6, 9]

使用count()方法可以避免重複計算和排序,因此相對更為高效。

五、使用標準庫heapq的nlargest()方法

當數組長度較大時,標準庫heapq的nlargest()方法可以提高排序性能。它可以在未排序的列表或可迭代對象中返回n個最大元素,其時間複雜度為O(nlogn)。

    import heapq
    
    lst = [3, 1, 4, 1, 5, 9, 2, 6, 5, 3, 5]
    sorted_lst = heapq.nlargest(len(lst), lst)
    print(sorted_lst)
    # 輸出結果:[9, 6, 5, 5, 5, 5, 4, 3, 3, 2, 1]

需要注意的是,nlargest()方法返回的結果是未排序的,如果需要對結果進行排序,可以用內置函數sorted()或list的sort()方法。

六、使用list的pop()方法和min()函數

對於需要在排序過程中找到最小值和次小值等操作,可以利用list的pop()方法和min()函數實現。pop()方法可以移除列表中的一個元素,並返回該元素,時間複雜度為O(1)。min()函數可以找到列表中最小值,時間複雜度為O(n)。

    lst = [3, 1, 4, 1, 5, 9, 2, 6, 5, 3, 5]
    sorted_lst = []
    while lst:
        sorted_lst.append(min(lst))
        lst.pop(lst.index(min(lst)))
    print(sorted_lst)
    # 輸出結果:[1, 1, 2, 3, 3, 4, 5, 5, 5, 6, 9]

需要注意的是,pop()方法需要在每一次循環中都查找最小值,因此的時間複雜度為O(n^2),對於較長的數組並不適用。

七、同時使用多個方法

對於複雜的排序問題,可以同時使用多個方法來提高性能。

    lst = [3, 1, 4, 1, 5, 9, 2, 6, 5, 3, 5]
    sorted_lst = []
    if len(lst) < 1000:
        sorted_lst = sorted(lst)
    else:
        arr = np.array(lst)
        sorted_arr = np.sort(arr)
        sorted_lst = sorted_lst.append(list(sorted_arr))
    print(sorted_lst)

以上代碼示例中,當數組長度小於1000時,使用內置函數sorted()進行排序,當數組長度大於等於1000時,使用numpy庫的sort()函數進行排序,既保證了排序性能,又保證了排序正確性。

總結

本篇文章介紹了Python中提高數組排序性能的七種方法,包括使用內置函數sorted()、sort()方法、numpy庫的sort()函數、list的count()方法、標準庫heapq的nlargest()方法、list的pop()方法和min()函數以及同時使用多個方法。不同的場景適用不同的方法,需要根據實際情況選擇。

原創文章,作者:小藍,如若轉載,請註明出處:https://www.506064.com/zh-tw/n/160921.html

(0)
打賞 微信掃一掃 微信掃一掃 支付寶掃一掃 支付寶掃一掃
小藍的頭像小藍
上一篇 2024-11-21 01:16
下一篇 2024-11-21 01:16

相關推薦

  • Python周杰倫代碼用法介紹

    本文將從多個方面對Python周杰倫代碼進行詳細的闡述。 一、代碼介紹 from urllib.request import urlopen from bs4 import Bea…

    編程 2025-04-29
  • Python計算陽曆日期對應周幾

    本文介紹如何通過Python計算任意陽曆日期對應周幾。 一、獲取日期 獲取日期可以通過Python內置的模塊datetime實現,示例代碼如下: from datetime imp…

    編程 2025-04-29
  • 如何查看Anaconda中Python路徑

    對Anaconda中Python路徑即conda環境的查看進行詳細的闡述。 一、使用命令行查看 1、在Windows系統中,可以使用命令提示符(cmd)或者Anaconda Pro…

    編程 2025-04-29
  • Python中引入上一級目錄中函數

    Python中經常需要調用其他文件夾中的模塊或函數,其中一個常見的操作是引入上一級目錄中的函數。在此,我們將從多個角度詳細解釋如何在Python中引入上一級目錄的函數。 一、加入環…

    編程 2025-04-29
  • Python列表中負數的個數

    Python列表是一個有序的集合,可以存儲多個不同類型的元素。而負數是指小於0的整數。在Python列表中,我們想要找到負數的個數,可以通過以下幾個方面進行實現。 一、使用循環遍歷…

    編程 2025-04-29
  • Python清華鏡像下載

    Python清華鏡像是一個高質量的Python開發資源鏡像站,提供了Python及其相關的開發工具、框架和文檔的下載服務。本文將從以下幾個方面對Python清華鏡像下載進行詳細的闡…

    編程 2025-04-29
  • Python字典去重複工具

    使用Python語言編寫字典去重複工具,可幫助用戶快速去重複。 一、字典去重複工具的需求 在使用Python編寫程序時,我們經常需要處理數據文件,其中包含了大量的重複數據。為了方便…

    編程 2025-04-29
  • 蝴蝶優化演算法Python版

    蝴蝶優化演算法是一種基於仿生學的優化演算法,模仿自然界中的蝴蝶進行搜索。它可以應用於多個領域的優化問題,包括數學優化、工程問題、機器學習等。本文將從多個方面對蝴蝶優化演算法Python版…

    編程 2025-04-29
  • Python程序需要編譯才能執行

    Python 被廣泛應用於數據分析、人工智慧、科學計算等領域,它的靈活性和簡單易學的性質使得越來越多的人喜歡使用 Python 進行編程。然而,在 Python 中程序執行的方式不…

    編程 2025-04-29
  • python強行終止程序快捷鍵

    本文將從多個方面對python強行終止程序快捷鍵進行詳細闡述,並提供相應代碼示例。 一、Ctrl+C快捷鍵 Ctrl+C快捷鍵是在終端中經常用來強行終止運行的程序。當你在終端中運行…

    編程 2025-04-29

發表回復

登錄後才能評論