csr_array#
- class scipy.sparse.csr_array(arg1, shape=None, dtype=None, copy=False, *, maxprint=None)[源代码]#
压缩稀疏行数组。
- 可以通过多种方式实例化它
- csr_array(D)
其中 D 是一个 2 维 ndarray
- csr_array(S)
使用另一个稀疏数组或矩阵 S (等效于 S.tocsr())
- csr_array((M, N), [dtype])
构造一个形状为 (M, N) 的空数组,dtype 是可选的,默认为 dtype='d'。
- csr_array((data, (row_ind, col_ind)), [shape=(M, N)])
其中
data
、row_ind
和col_ind
满足关系a[row_ind[k], col_ind[k]] = data[k]
。- csr_array((data, indices, indptr), [shape=(M, N)])
是标准的 CSR 表示形式,其中行 i 的列索引存储在
indices[indptr[i]:indptr[i+1]]
中,它们对应的值存储在data[indptr[i]:indptr[i+1]]
中。如果未提供形状参数,则数组维度将从索引数组推断。
备注
稀疏数组可以用于算术运算:它们支持加法、减法、乘法、除法和矩阵幂。
- CSR 格式的优点
高效的算术运算 CSR + CSR、CSR * CSR 等。
高效的行切片
快速矩阵向量积
- CSR 格式的缺点
慢速列切片运算(考虑使用 CSC)
对稀疏结构的更改开销很大(考虑使用 LIL 或 DOK)
- 规范格式
在每一行中,索引按列排序。
没有重复的条目。
示例
>>> import numpy as np >>> from scipy.sparse import csr_array >>> csr_array((3, 4), dtype=np.int8).toarray() array([[0, 0, 0, 0], [0, 0, 0, 0], [0, 0, 0, 0]], dtype=int8)
>>> row = np.array([0, 0, 1, 2, 2, 2]) >>> col = np.array([0, 2, 2, 0, 1, 2]) >>> data = np.array([1, 2, 3, 4, 5, 6]) >>> csr_array((data, (row, col)), shape=(3, 3)).toarray() array([[1, 0, 2], [0, 0, 3], [4, 5, 6]])
>>> indptr = np.array([0, 2, 3, 6]) >>> indices = np.array([0, 2, 2, 0, 1, 2]) >>> data = np.array([1, 2, 3, 4, 5, 6]) >>> csr_array((data, indices, indptr), shape=(3, 3)).toarray() array([[1, 0, 2], [0, 0, 3], [4, 5, 6]])
重复的条目会被加在一起
>>> row = np.array([0, 1, 2, 0]) >>> col = np.array([0, 1, 1, 0]) >>> data = np.array([1, 2, 4, 8]) >>> csr_array((data, (row, col)), shape=(3, 3)).toarray() array([[9, 0, 0], [0, 2, 0], [0, 4, 0]])
作为如何增量构建 CSR 数组的示例,以下代码片段从文本构建词-文档数组
>>> docs = [["hello", "world", "hello"], ["goodbye", "cruel", "world"]] >>> indptr = [0] >>> indices = [] >>> data = [] >>> vocabulary = {} >>> for d in docs: ... for term in d: ... index = vocabulary.setdefault(term, len(vocabulary)) ... indices.append(index) ... data.append(1) ... indptr.append(len(indices)) ... >>> csr_array((data, indices, indptr), dtype=int).toarray() array([[2, 1, 0, 0], [0, 1, 1, 1]])
- 属性:
- dtypedtype
数组的数据类型
- shape2 元组
数组的形状
- ndimint
维度数(始终为 2)
nnz
存储值的数量,包括显式的零。
size
存储值的数量。
- data
数组的 CSR 格式数据数组
- indices
数组的 CSR 格式索引数组
- indptr
数组的 CSR 格式索引指针数组
has_sorted_indices
索引是否已排序
has_canonical_format
数组/矩阵是否具有已排序的索引且没有重复项
T
转置。
方法
__len__
()arcsin
()逐元素反正弦。
arcsinh
()逐元素反双曲正弦。
arctan
()逐元素反正切。
arctanh
()逐元素反双曲正切。
argmax
([axis, out, explicit])返回沿轴的最大元素的索引。
argmin
([axis, out, explicit])返回沿轴的最小元素的索引。
asformat
(format[, copy])以传递的格式返回此数组/矩阵。
astype
(dtype[, casting, copy])将数组/矩阵元素转换为指定的类型。
ceil
()逐元素向上取整。
check_format
([full_check])检查数组/矩阵是否符合 CSR 或 CSC 格式。
conj
([copy])逐元素复共轭。
conjugate
([copy])逐元素复共轭。
copy
()返回此数组/矩阵的副本。
count_nonzero
([axis])非零条目的数量,等效于
deg2rad
()逐元素 deg2rad。
diagonal
([k])返回数组/矩阵的第 k 个对角线。
dot
(other)普通的点积
从数组/矩阵中删除零条目
expm1
()逐元素 expm1。
floor
()逐元素向下取整。
log1p
()逐元素 log1p。
max
([axis, out, explicit])返回数组/矩阵的最大值或沿轴的最大值。
maximum
(other)此数组/矩阵和另一个数组/矩阵之间的逐元素最大值。
mean
([axis, dtype, out])计算沿指定轴的算术平均值。
min
([axis, out, explicit])返回数组/矩阵的最小值或沿轴的最大值。
minimum
(other)此数组/矩阵和另一个数组/矩阵之间的逐元素最小值。
multiply
(other)按数组/矩阵、向量或标量进行逐点乘法。
nanmax
([axis, out, explicit])返回沿轴的最大值,忽略任何 Nan。
nanmin
([axis, out, explicit])返回沿轴的最小值,忽略任何 Nan。
nonzero
()数组/矩阵的非零索引。
power
(n[, dtype])此函数执行逐元素幂运算。
prune
()删除所有非零元素之后的空白空间。
rad2deg
()逐元素 rad2deg。
reshape
(self, shape[, order, copy])为稀疏数组/矩阵赋予新的形状,而不更改其数据。
resize
(*shape)将数组/矩阵就地调整为
shape
给定的维度rint
()逐元素四舍五入到最接近的整数。
setdiag
(values[, k])设置数组/矩阵的对角线或非对角线元素。
sign
()逐元素符号。
sin
()逐元素正弦。
sinh
()逐元素双曲正弦。
原地对该数组/矩阵的索引进行排序
返回一个包含排序索引的此数组/矩阵的副本
sqrt
()逐元素求平方根。
sum
([axis, dtype, out])对给定轴上的数组/矩阵元素求和。
通过将重复项相加来消除重复条目
tan
()逐元素求正切。
tanh
()逐元素求双曲正切。
toarray
([order, out])返回此稀疏数组/矩阵的密集 ndarray 表示。
tobsr
([blocksize, copy])将此数组/矩阵转换为块稀疏行格式。
tocoo
([copy])将此数组/矩阵转换为坐标格式。
tocsc
([copy])将此数组/矩阵转换为压缩稀疏列格式。
tocsr
([copy])将此数组/矩阵转换为压缩稀疏行格式。
todense
([order, out])返回此稀疏数组的密集表示。
todia
([copy])将此数组/矩阵转换为稀疏对角线格式。
todok
([copy])将此数组/矩阵转换为键字典格式。
tolil
([copy])将此数组/矩阵转换为列表的列表格式。
trace
([offset])返回稀疏数组/矩阵对角线上的和。
transpose
([axes, copy])反转稀疏数组/矩阵的维度。
trunc
()逐元素截断。
__getitem__
__mul__