import numpy as np
matrix = [[1,2,3,4],[5,6,7,8],[4,5,6,7],[4,3,2,1]]]
#比如取第一行和第三行
matrix = np.array(matrix)#转换为array格式
matrix[[0,2],:]
print(matrix)
out : [[1,2,3,4],[4,5,6,7]]
import numpy as np
matrix = [[1,2,3,4],[5,6,7,8],[4,5,6,7],[4,3,2,1]]]
#比如取第一行和第三行
matrix = np.array(matrix)#转换为array格式
matrix[[0,2],:]
print(matrix)
out : [[1,2,3,4],[4,5,6,7]]