scipy.sparse 稀疏矩陣


 from 博客園(華夏35度)http://www.cnblogs.com/zhangchaoyang 作者:Orisun

本文主要圍繞scipy中的稀疏矩陣展開,也會介紹幾種scipy之外的稀疏矩陣的存儲方式。

dok_matrix

繼承自dict,key是(row,col)構成的二元組,value是非0元素。

優點:

  1. 非常高效地添加、刪除、查找元素
  2. 轉換成coo_matrix很快

缺點:

  1. 繼承了dict的缺點,即內存開銷大
  2. 不能有重復的(row,col)

適用場景:

  1. 加載數據文件時使用dok_matrix快速構建稀疏矩陣,然后轉換成其他形式的稀疏矩陣

coo_matrix

如上圖,構造coo_matrix需要3個等長的數組,values數組存放矩陣中的非0元素,row indices存放非0元素的行坐標,column indices存放非0元素的列坐標。

優點:

  1. 容易構造
  2. 可以快速地轉換成其他形式的稀疏矩陣
  3. 支持相同的(row,col)坐標上存放多個值

缺點:

  1. 構建完成后不允許再插入或刪除元素
  2. 不能直接進行科學計算和切片操作

適用場景:

  1. 加載數據文件時使用coo_matrix快速構建稀疏矩陣,然后調用to_csr()、to_csc()、to_dense()把它轉換成CSR或稠密矩陣

csr_matrix

csr_matrix同樣由3個數組組成,values存儲非0元素,column indices存儲非0元素的列坐標,row offsets依次存儲每行的首元素在values中的坐標,如果某行全是0則對應的row offsets值為-1(我猜的)。

優點:

  1. 高效地按行切片
  2. 快速地計算矩陣與向量的內積
  3. 高效地進行矩陣的算術運行,CSR + CSR、CSR * CSR等

缺點:

  1. 按列切片很慢(考慮CSC)
  2. 一旦構建完成后,再往里面添加或刪除元素成本很高

csc_matrix

跟csr_matrix剛好反過來。

bsr_matrix

跟CSR/CSC很相近,尤其適用於稀疏矩陣中包含稠密子矩陣的情況。在解決矢量值有限元離散(vector-valued finite element discretizations)這類問題中BSR比CSR/CSC更高效。

dia_matrix

對角線存儲法,按對角線方式存,列代表對角線,行代表行。省略全零的對角線。(從左下往右上開始:第一個對角線是零忽略,第二個對角線是5,6,第三個對角線是零忽略,第四個對角線是1,2,3,4,第五個對角線是7,8,9,第六第七個對角線忽略)。[3]

這里行對應行,所以5和6是分別在第三行第四行的,前面補上無效元素*。如果對角線中間有0,存的時候也需要補0。

適用場景:

  1. 如果原始矩陣就是一個對角性很好的矩陣那壓縮率會非常高,比如下圖,但是如果是隨機的那效率會非常糟糕。

lil_matrix

內部結構是個二維數組:[[(col,value)]],第一行對應原矩陣的一行(可以快速地定位到行),行內按列編號排序好(通過折半查找可以快速地定位到列),同樣只存儲非0元素。

優點:

  1. 快速按行切片
  2. 高效地添加、刪除、查找元素

缺點:

  1. 按列切片很慢(考慮CSC)
  2. 算術運算LIL+LIL很慢(考慮CSR或CSC)
  3. 矩陣和向量內和解很慢(考慮CSR或CSC)

適用場景:

  1. 加載數據文件時使用lil_matrix快速構建稀疏矩陣,然后調用to_csr()、to_csc()把它轉換成CSR/CSC進行后續的矩陣運算
  2. 非0元素非常多時,考慮使用coo_matrix(我個人是這樣理解的,lil_matrix用一個二維數組搞定,二維數組占用的是連續的內存空間,如果非0元素非常多就要申請一塊非常大的連續的內存空間,這樣性能很差。而coo_matrix畢竟是使用的3個一維數組,對連續內存空間的要求沒那么高)

 ELLPACK (ELL)

用兩個和原始矩陣相同行數的矩陣來存:第一個矩陣存的是列號,第二個矩陣存的是數值,行號就不存了,用自身所在的行來表示;這兩個矩陣每一行都是從頭開始放,如果沒有元素了就用個標志比如*結束。 上圖中間矩陣有誤,第三行應該是  0 2 3。

注:這樣如果某一行很多元素,那么后面兩個矩陣就會很胖,其他行結尾*很多,浪費。可以存成數組,比如上面兩個矩陣就是:

0 1 * 1 2 * 0 2 3 * 1 3 *

1 7 * 2 8 * 5 3 9 * 6 4 *

但是這樣要取一行就比較不方便了。

Hybrid (HYB) ELL + COO

為了解決ELL中提到的,如果某一行特別多,造成其他行的浪費,那么把這些多出來的元素(比如第三行的9,其他每一行最大都是2個元素)用COO單獨存儲。

skyline matrix storage

沒看明白,自行wiki。

適用場景:

  1. 非常適合於稀疏矩陣的Cholesky分解或LU分解,這兩種分解都是用來解線性方程組的。

行列雙索引

這是自己實現的一種存儲方式,分別按行和按列建立dict(dict中的key是行號或列號),這樣按下標查找元素很快,但犧牲了空間。為了挽回空間上的犧牲,我們采用二進制來存儲dict中的value。按下標查找元素時,根據行號定位到相應的value,value反序列化后轉成dict,該dict的key是列號。

上面的代碼中做二進制序列化時用到了struck.pack,來個小例子看下序列化能省多少內存。

優點:

  1. 高效地動態添加元素
  2. 高效地按下標查找元素
  3. 高效地按行切片和按列切片

缺點:

  1. 不支持刪除元素
  2. 內存占用略大

選擇稀疏矩陣存儲格式的經驗

  1. DIA和ELL格式在進行稀疏矩陣-矢量乘積(sparse matrix-vector products)時效率最高,所以它們是應用迭代法(如共軛梯度法)解稀疏線性系統最快的格式
  2. COO格式常用於從文件中進行稀疏矩陣的讀寫,如matrix market即采用COO格式,而CSR格式常用於讀入數據后進行稀疏矩陣計算


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM