Skip to content
Snippets Groups Projects
Select Git revision
  • 3c5b5eb1721151687aff18a7505b59eaa99cc31e
  • master default protected
2 results

utilities.py

Blame
  • Code owners
    Assign users and groups as approvers for specific file changes. Learn more.
    utilities.py 597 B
    """Collection of commonly used functions."""
    
    import sys
    import csv
    
    
    ERRORS = {
        'not_csv_suffix': '{fname}: File must end in .csv',
        }
    
    
    def collection_to_csv(collection, num=None):
        """
        Write collection of items and counts in csv format.
    
        Parameters
        ----------
        collection : collections.Counter
            Collection of items and counts
        num : int
            Limit output to N most frequent items
        """
        collection = collection.most_common()
        if num is None:
            num = len(collection)
        writer = csv.writer(sys.stdout)
        writer.writerows(collection[0:num])