import numpy as np a = [ [1,1], [1,1] ]b= [ [10,100], [1000,10000] ]a = np.array(a)print(a[a == 1])
Standard input is empty
[1 1 1 1]
The brand new service which powers Ideone!
Widget for compiling and running the source code in a web browser!