Python程式判斷給定矩陣是否為稀疏矩陣


矩陣是一個矩形陣列,其中數字按行和列排列。它被稱為m X n矩陣,其中m和n是維度。

如果矩陣中非零元素的數量遠少於零元素的數量,則稱其為稀疏矩陣。

[0, 0, 3, 0, 0]
[0, 1, 0, 0, 6]
[1, 0, 0, 9, 0]
[0, 0, 2, 0, 0]

上面的矩陣是4X5矩陣,其中大部分數字為零。只有少數元素是非零的,因此我們可以稱之為稀疏矩陣。

要檢查給定的矩陣是否為稀疏矩陣,我們需要比較元素總數和零元素的數量。如果零元素的數量超過矩陣中元素數量的一半,則可以將給定矩陣稱為稀疏矩陣。

(m * n)/2

讓我們討論一下判斷給定矩陣是否為稀疏矩陣的不同方法。

使用For迴圈

使用for迴圈,我們可以輕鬆迭代Python中的陣列元素。

示例

首先,我們將迭代矩陣行並計算每一行中零的個數。然後將計數值儲存在計數器變數中。

之後,我們將計數器變數中的值與矩陣中元素數量的一半進行比較,以確定給定的矩陣是否為稀疏矩陣。

def isSparse(array, m, n):
   counter = 0
   # Count number of zeros
   for i in range(0, m):
      for j in range(0, n):
         if (array[i][j] == 0):
            counter = counter + 1
   return (counter > ((m * n) // 2))

arr = [[0, 0, 3],
       [0, 0, 0],
       [1, 8, 0]]

print("The original matrix: ")
for row in arr:
   print(row)
print()

# check if the given matrix is sparse matrix or not
if (isSparse(arr, len(arr), len(arr[0]))):
   print("The given matrix is a sparse matrix")
else:
   print("The given matrix is not a sparse matrix")

輸出

The original matrix: 
[0, 0, 3]
[0, 0, 0]
[1, 8, 0]

The given matrix is a sparse matrix

上述矩陣是一個稀疏矩陣。

示例

在這個例子中,我們將使用list.count()方法來計算迴圈中每一行中零元素的個數,並將計數儲存在計數器變數中。

def isSparse(array, m, n):
   counter = 0
   # Count number of zeros
   for i in array:
      counter += i.count(0)
   return (counter > ((m * n) // 2))

arr = [[0, 0, 3],
       [0, 0, 0],
       [1, 8, 0]]

print("The original matrix: ")
for row in arr:
   print(row)
print()

# check if the given matrix is sparse matrix or not
if (isSparse(arr, len(arr), len(arr[0]))):
   print("The given matrix is a sparse matrix")
else:
   print("The given matrix is not a sparse matrix")

輸出

The original matrix: 
[0, 0, 3]
[0, 0, 0]
[1, 8, 0]

The given matrix is a sparse matrix

使用SciPy庫

透過在Python中使用SciPy庫,我們可以建立稀疏矩陣。在下面的例子中,我們使用了csr_matrix()函式以壓縮稀疏行格式建立稀疏矩陣。

而issparse()函式用於檢查給定物件是否為稀疏矩陣。

示例

首先,我們將使用巢狀列表建立一個數組,然後使用csr_matrix()方法將其轉換為稀疏矩陣。

from scipy.sparse import issparse, csr_matrix
arr = [[0, 0, 3],
       [0, 0, 0],
       [1, 8, 0]]

matrix = csr_matrix(arr)

print("The original matrix: ")
print(matrix)
print()

# check if the given matrix is sparse matrix or not
if (issparse(matrix)):
   print("The given matrix is a sparse matrix")
else:
   print("The given matrix is not a sparse matrix")

輸出

The original matrix: 
  (0, 2)	3
  (2, 0)	1
  (2, 1)	8

The given matrix is a sparse matrix

csr_matrix()方法只將資料點(非零元素)儲存在記憶體中。

注意 - issparse()方法與輸入矩陣有多少個元素無關。相反,它檢查給定物件是否為spmatrix的例項。

更新於:2023年5月15日

498 次瀏覽

開啟你的職業生涯

完成課程獲得認證

開始學習
廣告