def gen_kwargs(format):
from sparse._utils import convert_format
format = convert_format(format)
if format == "gcxs":
return [{"compressed_axes": c} for c in [(0,), (1,)]]
return [{}]
def asformat(self, format, **kwargs):
"""
Convert this sparse array to a given format.
Parameters
----------
format : str
A format string.
Returns
-------
out : SparseArray
The converted array.
Raises
------
NotImplementedError
If the format isn't supported.
"""
from sparse._utils import convert_format
format = convert_format(format)
ret = None
if format == "coo":
ret = self.tocoo()
elif format == "dok":
ret = self.todok()
elif format == "csr":
ret = CSR(self)
elif format == "csc":
ret = CSC(self)
elif format == "gcxs":
compressed_axes = kwargs.pop("compressed_axes",
self.compressed_axes)
return self.change_compressed_axes(compressed_axes)
if len(kwargs) != 0:
raise TypeError(f"Invalid keyword arguments provided: {kwargs}")
if ret is None:
raise NotImplementedError(
f"The given format is not supported: {format}")
return ret
def asformat(self, format, **kwargs):
"""
Convert this sparse array to a given format.
Parameters
----------
format : str
A format string.
Returns
-------
out : SparseArray
The converted array.
Raises
------
NotImplementedError
If the format isn't supported.
"""
from sparse._utils import convert_format
format = convert_format(format)
if format == "gcxs":
from .._compressed import GCXS
return GCXS.from_coo(self, **kwargs)
elif len(kwargs) != 0:
raise TypeError(f"Invalid keyword arguments provided: {kwargs}")
if format == "coo":
return self
if format == "dok":
from .._dok import DOK
return DOK.from_coo(self, **kwargs)
return self.asformat("gcxs", **kwargs).asformat(format, **kwargs)
def asformat(self, format, **kwargs):
"""
Convert this sparse array to a given format.
Parameters
----------
format : str
A format string.
Returns
-------
out : SparseArray
The converted array.
Raises
------
NotImplementedError
If the format isn't supported.
"""
from sparse._utils import convert_format
format = convert_format(format)
if format == "dok":
return self
if format == "coo":
from ._coo import COO
if len(kwargs) != 0:
raise ValueError(f"Extra kwargs found: {kwargs}")
return COO.from_iter(
self.data,
shape=self.shape,
fill_value=self.fill_value,
dtype=self.dtype,
)
return self.asformat("coo").asformat(format, **kwargs)