数组之间可以通过索引直接取值

a = np.arrapy([3,6,8,9,10])
#取a的索引为1和4
a_index = np.array([1,4])
a[a_index]
打印的结果就是:array([6,10])

猜你喜欢

转载自blog.csdn.net/weixin_44274975/article/details/88695103