mean_variance_axis#

sklearn.utils.sparsefuncs.mean_variance_axis(X, axis, weights=None, return_sum_weights=False)[source]#

沿 CSR 或 CSC 矩阵的轴计算均值和方差。

参数:

Xsparse matrix of shape (n_samples, n_features): 输入数据。可以是 CSR 或 CSC 格式。
axis{0, 1}: 计算均值和方差的轴。
weightsndarray of shape (n_samples,) or (n_features,), default=None: 如果 axis 设置为 0，形状为 (n_samples,)；如果 axis 设置为 1，形状为 (n_features,)。如果设置为 None，则样本权重相等。

0.24 版本新增。
return_sum_weightsbool, default=False: 如果为 True，则返回每个特征（如果 axis=0）或每个样本（如果 axis=1）的权重总和。

0.24 版本新增。

返回:

meansndarray of shape (n_features,), dtype=floating: 按特征计算的均值。
variancesndarray of shape (n_features,), dtype=floating: 按特征计算的方差。
sum_weightsndarray of shape (n_features,), dtype=floating: 如果 return_sum_weights 为 True 则返回。

示例

>>> from sklearn.utils import sparsefuncs
>>> from scipy import sparse
>>> import numpy as np
>>> indptr = np.array([0, 3, 4, 4, 4])
>>> indices = np.array([0, 1, 2, 2])
>>> data = np.array([8, 1, 2, 5])
>>> scale = np.array([2, 3, 2])
>>> csr = sparse.csr_matrix((data, indices, indptr))
>>> csr.todense()
matrix([[8, 1, 2],
        [0, 0, 5],
        [0, 0, 0],
        [0, 0, 0]])
>>> sparsefuncs.mean_variance_axis(csr, axis=0)
(array([2.  , 0.25, 1.75]), array([12.    ,  0.1875,  4.1875]))

mean_variance_axis#

本页