jamespeterschinner/text_array.py

## text_array.py
import numpy as np

def horizontal_index(n, offset=0):
    """Create horizontal index with vertically aligned numbers"""
    index = [' ' * offset for _ in range(len(str(n)))]
    for number in map(str, range(n)):
        for idx, n in enumerate(number):
            index[idx] += n + '|'
    pad = max(map(len, index))
    return '\n'.join(i.rjust(pad, ' ') for i in reversed(index))


def vertical_index(n):
    """Generate row numbers for an index"""
    width = len(str(n))
    index = map(str, range(n))
    for ind in (i.rjust(width, ' ') + '|' for i in index):
        yield '\n' + ind


class TextArray(object):

    def __init__(self, text):
        # Pad out each row to the max len row
        rows = text.split('\n')
        max_len = max(map(len, rows))
        self.array = np.array([list(str(row).ljust(max_len, ' ')) for row in rows])

    def __getitem__(self, index):
        row, column = index
        return self.array[row, column]

    def __setitem__(self, key, value):
        row, column = key
        self.array[row, column] = value

    def __repr__(self):
        """Create a nice console display"""
        row_index = vertical_index(self.array.shape[0])
        column_index = horizontal_index(self.array.shape[1], len(str(self.array.shape[0])) + 1)
        result = ''.join(row + ' '.join(self.array[idx]) for idx, row in enumerate(row_index))
        return f'{column_index}' \
               f'{result}'
	import numpy as np

	def horizontal_index(n, offset=0):
	"""Create horizontal index with vertically aligned numbers"""
	index = [' ' * offset for _ in range(len(str(n)))]
	for number in map(str, range(n)):
	for idx, n in enumerate(number):
	index[idx] += n + '\|'
	pad = max(map(len, index))
	return '\n'.join(i.rjust(pad, ' ') for i in reversed(index))


	def vertical_index(n):
	"""Generate row numbers for an index"""
	width = len(str(n))
	index = map(str, range(n))
	for ind in (i.rjust(width, ' ') + '\|' for i in index):
	yield '\n' + ind


	class TextArray(object):

	def __init__(self, text):
	# Pad out each row to the max len row
	rows = text.split('\n')
	max_len = max(map(len, rows))
	self.array = np.array([list(str(row).ljust(max_len, ' ')) for row in rows])

	def __getitem__(self, index):
	row, column = index
	return self.array[row, column]

	def __setitem__(self, key, value):
	row, column = key
	self.array[row, column] = value

	def __repr__(self):
	"""Create a nice console display"""
	row_index = vertical_index(self.array.shape[0])
	column_index = horizontal_index(self.array.shape[1], len(str(self.array.shape[0])) + 1)
	result = ''.join(row + ' '.join(self.array[idx]) for idx, row in enumerate(row_index))
	return f'{column_index}' \
	f'{result}'