def vstack(lst): """ Vstack that considers sparse matrices :param lst: :return: """ return sp.vstack(lst) if sp and isinstance(lst[0], sp.sparse.csr.csr_matrix) else np.vstack(lst)