o Uݢg<@sjddlmZddlmZddlZddlZddlm Z m Z dd d Z dddZ ddddZ ddddZdS)) annotations)IterableN) ensure_binary ensure_strpathstrreturn list[str]cCsttj|ddjS)zGets the headers of csv files with # as comments. Args: path (Union[str, bytes]): path of the csv Returns: List[str]: List of column names of probe_set.csv #)comment)listpdread_csvcolumns)rr`/oak/stanford/groups/akundaje/marinovg/programs/cellranger-9.0.1/lib/python/cellranger/csv_io.pyload_csv_columnnamess rcsv_file str | bytes%np.ndarray[int, np.dtype[np.object_]]cCs,ttjt|dgdtiddj}|S)NrF)usecolsZ convertersZ na_filter)npZ atleast_1dr rrrvaluesZsqueeze)rZrownamesrrrload_csv_rownamessr feature_ints Iterable[int]NonecCsdt|d#}|dur|t|d|D] }||dqWddS1s+wYdS)zWrite a list of ints to a file.wN )openwriter)rrheaderZoutfirrrwrite_target_features_csv$s "r#r!str | int | Nonelist[int | str]cKs\|durtjt|fi|}||}|Stjt|fdgdd|}|d}|S)zRead feature IDs from a file.Nr)rr!)r rrtolist)rr!kwargsZ feature_idsrrrread_target_features_csv-s  r()rrrr )rrrr)N)rrrrrr)rrr!r$rr%) __future__rcollections.abcrZnumpyrZpandasr Zsixrrrrr#r(rrrrs