def cast_list_float(_list):
return [float(value) for value in _list]
if __name__ == "__main__":
data = {}
# データ要素を取り出す
for table in TABLE_NAMES:
columns = db_api.get_column_names(table)
for column in columns:
_list = db_api.pick_column(table, column)
if column in ab_normal_column:
_list = convert_abnormal_column_list(_list)
data[f"{column}_sum"] = caluculator.create_sum_list(_list)
data[f"{column}_average"] = caluculator.create_average_list(
_list)
elif column == "date":
_list = [x.weekday() for x in _list]
data["weekday"] = _list
elif column in skip_column:
pass
else:
_list = cast_list_float(_list)
data[column] = _list
keys = list(data.keys())
combine_list = caluculator.create_combination_list(keys, 2)
for combi in combine_list:
try:
grapher.scatter_plot_with_corrcoef(data[combi[0]], combi[0],
data[combi[1]], combi[1])
def test_create_average_list(self):
calculator = Calculator()
_input = [[1, 2, 3], [4, 5, 6]]
_output = [2.0, 5.0]
self.assertListEqual(calculator.create_average_list(_input), _output)