mardi 20 août 2019

Python elif optimization problems

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