how to optimize this code?? how to optimize this code as short as possible. I am a beginner please help me out.CONDITIONS: - arr must be a valid 2D array with valid numbers only. - data can be : 'column' or 'row' - value can be : 'min' or 'max' or 'mean' or 'median' - array must be a valid 2D Array with Integer / Float ONLY
def get_math_value(array, data, value):
if data == 'row' and value == 'min':
arr = array
min_row = list(map(min, arr))
print(min_row)
elif data == 'row' and value == 'max':
arr = array
max_row = list(map(min, arr))
print(max_row)
elif data == 'row' and value == 'mean':
arr = array
mean_row = np.mean(arr, axis=1)
print(mean_row)
elif data == 'row' and value == 'median':
arr = array
median_row = np.median(arr, axis=1)
print(median_row)
elif data == 'column' and value == 'min':
arr = array
min_column = list(map(min, zip(*arr)))
print(min_column)
elif data == 'column' and value == 'max':
arr = array
max_column = list(map(max, zip(*arr)))
print(max_column)
elif data == 'column' and value == 'mean':
arr = array
mean_column = np.mean(arr, axis=0)
print(mean_column)
elif data == 'column' and value == 'median':
arr = array
median_column = np.median(arr, axis=0)
print(median_column)
else:
print('[]')
Aucun commentaire:
Enregistrer un commentaire