from pathlib import Path from typing import Optional, Sequence, Union from pandas.compat._optional import import_optional_dependency from pandas.core.dtypes.inference import is_list_like from pandas.core.api import DataFrame from pandas.io.common import stringify_path def read_spss( path: Union[str, Path], usecols: Optional[Sequence[str]] = None, convert_categoricals: bool = True, ) -> DataFrame: """ Load an SPSS file from the file path, returning a DataFrame. .. versionadded:: 0.25.0 Parameters ---------- path : str or Path File path. usecols : list-like, optional Return a subset of the columns. If None, return all columns. convert_categoricals : bool, default is True Convert categorical columns into pd.Categorical. Returns ------- DataFrame """ pyreadstat = import_optional_dependency("pyreadstat") if usecols is not None: if not is_list_like(usecols): raise TypeError("usecols must be list-like.") else: usecols = list(usecols) # pyreadstat requires a list df, _ = pyreadstat.read_sav( stringify_path(path), usecols=usecols, apply_value_formats=convert_categoricals ) return df