A table-like object that can be sliced like a numpy array. Pure python.

array_table.py

class Table(list):
    """
    a table object that can be sliced like a numpy array
    """

    def __init__(self, *args, has_header=True):
        self.has_header = has_header

        if has_header:
            self.header_index = {}
            self.header = args[0][0]
            self.__make_header_index()

        list.__init__(self, *args)

    def __getitem__(self, item):
        try:
            return list.__getitem__(self, item)
        except TypeError:
            rows, cols = item
            return [row[cols] for row in self[rows]]

    def __make_header_index(self):
        # set up a column heading index for lookups
        for idx, col_name in enumerate(self.header):
            self.header_index[col_name] = idx

    def column_unique(self, col_name):
        if self.has_header:
            # use a string to get the index from the header_index
            return set(self[1:, self.header_index[col_name]])
        else:
            # use an integer index
            return set(self[0:, col_name])

    def column(self, col_name):
        if self.has_header:
            # use a string to get the index from the header_index
            return self[1:, self.header_index[col_name]]
        else:
            # use an integer index
            return self[0:, col_name]