python 取数组中的第n行或者第n列

a = np.array([[1, 2], [3, 4], [5, 6], [7, 8]])
a[:, 0]
Out[4]: array([1, 3, 5, 7])
a[:, 1]
Out[5]: array([2, 4, 6, 8])
a[0]
Out[6]: array([1, 2])
a[1]
Out[7]: array([3, 4])

 

 

 

a1 = np.array([[10, 7, 4], [3, 2, 1]])
b = np.median(a1)
b
Out[17]: 3.5
a1 = np.array([[1, 3, 4], [3, 2, 1]])
b = np.median(a1)
b
Out[20]: 2.5
a1[a1>2] = 0
a1
Out[22]: 
array([[1, 0, 0],
       [0, 2, 1]])