128 lines
3.8 KiB
Python
128 lines
3.8 KiB
Python
#########################################################
|
|
## @file : mtable.py
|
|
## @desc : pandas is to large,
|
|
## so i implement a simple table data structure
|
|
## by my self
|
|
## @create : 2020/06/25
|
|
## @author : Chengan
|
|
## @email : douboer@gmail.com
|
|
#########################################################
|
|
|
|
class mTable():
|
|
|
|
def __init__(self, data=[], index=[], columns=[]):
|
|
self._data = data
|
|
self._index = index
|
|
self._columns = columns
|
|
self._width = 10
|
|
|
|
def dataframe(self, data=[], index=[], columns=[]):
|
|
self._data = data
|
|
self._index = index
|
|
self._columns = columns
|
|
|
|
def get_iat(self, row: int, column: int):
|
|
""" just like pandas iat"""
|
|
return self._data[row][column] if len(self._data)>0 else None
|
|
|
|
def get_num_rows(self):
|
|
return len(self._data) if len(self._data)>0 else 0
|
|
|
|
def get_num_columns(self):
|
|
return len(self._columns) if len(self._columns)>0 else 0
|
|
|
|
def reverse(self):
|
|
self._data.reverse()
|
|
|
|
def set_data(self, data=[]):
|
|
""" get table content"""
|
|
self._data = data
|
|
|
|
def get_data(self):
|
|
""" get table content"""
|
|
return self._data
|
|
|
|
def get_index(self):
|
|
""" just like pandas index"""
|
|
return self._index
|
|
|
|
def get_columns(self):
|
|
""" just like pandas columns"""
|
|
return self._columns
|
|
|
|
def set_repr_width(self, width):
|
|
self._width = width
|
|
|
|
def __repr__(self):
|
|
""" print table """
|
|
w = self._width
|
|
s = ''.ljust(w-1,' ')
|
|
for c in self._columns: s += c.ljust(w,' ')
|
|
s += '\n'
|
|
len_index = len(self._index)
|
|
for i, r in enumerate(self._data):
|
|
s += self._index[i].ljust(w,' ') if i<len_index else str(i).ljust(w,' ')
|
|
for c in r:
|
|
s += str(c).ljust(w,' ')
|
|
s += '\n'
|
|
return s
|
|
|
|
def __next__(self):
|
|
raise NotImplementedError
|
|
|
|
def __iter__(self):
|
|
return self
|
|
|
|
if __name__=='__main__':
|
|
|
|
"""
|
|
# move to unitest.kman.py
|
|
print('\n---------------------')
|
|
data = [['Ohiox','Ohio','Ohio','Nevada','Nevada'],
|
|
[2000,2001,2002,2001,2002],
|
|
[1.5,1.7,3.6,2.4,2.9],
|
|
[1.5,1.7,3.6,2.4,2.9],
|
|
]
|
|
|
|
mt = mTable(data,
|
|
index = ['one', 'two','three','four','five'],
|
|
columns = ['year','state','pop','debt','xx'])
|
|
|
|
print('== frame\n', mt)
|
|
print('== frame.iat[0, 0]\n', mt.get_iat(0, 0))
|
|
print('== frame.iat[1, 1]\n', mt.get_iat(1, 1))
|
|
print('== frame.iat[2, 2]\n', mt.get_iat(2, 2))
|
|
print('== frame.iat[2, 1]\n', mt.get_iat(2, 1))
|
|
print('== frame.shape[0]\n', mt.get_num_rows())
|
|
print('== frame.shape[1]\n', mt.get_num_columns())
|
|
print('== frame.columns\n', mt.get_columns())
|
|
print('== frame.index\n', mt.get_index())
|
|
|
|
mt = mTable(data,
|
|
columns = ['year','state','pop','debt','xx'])
|
|
print('== frame\n', mt)
|
|
print('== frame.iat[2, 1]\n', mt.get_iat(2, 1))
|
|
print('== frame.shape[0]\n', mt.get_num_rows())
|
|
print('== frame.shape[1]\n', mt.get_num_columns())
|
|
print('== frame.columns\n', mt.get_columns())
|
|
print('== frame.index\n', mt.get_index())
|
|
|
|
mt = mTable(data)
|
|
mt.set_repr_width(20)
|
|
print('== frame\n', mt)
|
|
print('== frame.iat[2, 1]\n', mt.get_iat(2, 1))
|
|
print('== frame.shape[0]\n', mt.get_num_rows())
|
|
print('== frame.shape[1]\n', mt.get_num_columns())
|
|
print('== frame.columns\n', mt.get_columns())
|
|
print('== frame.index\n', mt.get_index())
|
|
|
|
mt = mTable()
|
|
print('== frame\n', mt)
|
|
print('== frame.iat[2, 1]\n', mt.get_iat(2, 1))
|
|
print('== frame.shape[0]\n', mt.get_num_rows())
|
|
print('== frame.shape[1]\n', mt.get_num_columns())
|
|
print('== frame.columns\n', mt.get_columns())
|
|
print('== frame.index\n', mt.get_index())
|
|
"""
|
|
pass
|