def get_most_read_or_features(
        amb_sams: list,
        counts: dict) -> list:
    """
    Get the samples that have the most
    counts (reads of features).

    Parameters
    ----------
    amb_sams : list
        Sample IDs.
    counts : dict
        Count per Sample ID.

    Returns
    -------
    cur_best_samples : list
        Selected samples (with max reads/features).
    """
    cur_counts = dict((amb_sam, counts[amb_sam]) for amb_sam in amb_sams)
    count_max = max(cur_counts.values())
    cur_best_samples = [amb_sam for amb_sam in amb_sams
                        if cur_counts[amb_sam] == count_max]
    return cur_best_samples