当前位置：网站首页>numpy中选择特定两行

numpy中选择特定两行

2022-07-19 05:15:00 【lyz_fish】

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]]

版权声明
本文为[lyz_fish]所创，转载请带上原文链接，感谢
https://blog.csdn.net/qq_44881486/article/details/121441643