Python, Machine Learning, Numpy, Matplotlib.
return indices of maximal values column_argmax = np.argmax(X, axis=0) # note that the "index" here is actually the row the maximum occurs for each column print("Matrix X: \n") print(X) print("\n Maximum value in X: \n") print(all_max) print("\n Column-wise max of X: \n") print(column_max) print("\n