def __exelwriter(
self, csvfile
): # creates same named as csvfile xlsx and write data to it
wb = Workbook(self.__reportsdirectory + csvfile[:-4] + '.xlsx')
ws = wb.create_sheet("Sheet")
with open(csvfile, 'rt', newline="") as f:
reader = csv.reader(f, delimiter=';')
for subarray in reader:
subarray.append(csvfile[:-3])
ws.append(subarray)
wb.save(self.__reportsdirectory + csvfile[:-3] + '.xlsx')
def save(self, sa):
"""
:param sa:
"""
ext = os.path.splitext(sa)[1][1:].strip().lower()
if self.nparray is None:
raise ValueError
t = self.nparray[:, 0]
r_a = self.nparray[:, 1]
if sa:
if ext == 'dat':
r_n = r_a / 10
n_nparray = np.column_stack((t, r_n))
np.savetxt(sa, n_nparray, delimiter='\t', fmt=['%d', '%.3f'])
elif ext == 'xslx':
try:
from xlsxwriter import Workbook
except ImportError:
raise XLSWImportError
wb = Workbook(sa)
ws = wb.add_worksheet("summary_distances")
ws.write_row(0, 0, ('Time, ps', 'COM, \u212b'))
ws.write_column(1, 0, t)
ws.write_column(1, 1, r_a)
wb.close()
elif ext == 'xls':
try:
from xlwt import Workbook as Workbook
except ImportError:
raise XLWTImportError
wb = Workbook()
ws = wb.add_sheet("summary_distances")
ws.write(0, 0, 'Time, ps')
ws.write(0, 1, 'COM, \u212b')
for i, t_i in enumerate(t, start=1):
ws.write(i, 0, t_i)
for j, r_a_j in enumerate(r_a, start=1):
ws.write(j, 1, r_a_j)
wb.save(sa)
else:
raise BadExtError
