import numpy as np
x = np.array([[ 0, 1, 2],[ 3, 4, 5],[ 6, 7, 8],[ 9, 10, 11]])
print( '我们的数组是:' )
print (x)
print ('\n')
rows = np.array([[0,0],[3,3]])
cols = np.array([[0,2],[0,2]])
y = x[rows,cols]
print ('y1:' )
print (y)
rows1 = np.array([[0,0,3,3]])
cols1 = np.array([[0,2,0,2]])
y1 = x[rows1,cols1]
print ('y1:' )
print (y1)
我们的数组是:
[[ 0 1 2]
[ 3 4 5]
[ 6 7 8]
[ 9 10 11]]
y1:
[[ 0 2]
[ 9 11]]
y1:
[[ 0 2 9 11]]