Question

How to find the indexes of a matrix elements that are equal or greater than column average in Python

Answer 1

import numpy as np

arr = np.array([[3, 4, 1, 1], 
                [2, 5, 8, 7], 
                [9, 6, 2, 2]])

arr_true_false = arr > arr.mean(axis = 0)

indexes = np.where(arr_true_false == True)

print(list(zip(indexes[0], indexes[1])))




'''
run:

[(1, 2), (1, 3), (2, 0), (2, 1)]

'''

 

70+ SQL courses for beginners and professionals