import numpy >>> a = numpy.array([[1,2,3,4,5],[6,7,8,9,10],[1,2,3,4,5],[6,7,8,9,10]]) >>> a array([[ 1, 2, 3, 4, 5], [ 6, 7, 8, 9, 10], [ 1, 2, 3, 4, 5], [ 6, 7, 8, 9, 10]]) >>> a[...,2] array([3, 8, 3, 8]) >>> a[...,:2] array([[1, 2], [6, 7], [1, 2], [6, 7]]) >>> a[...,::2] array([[ 1, 3, 5], [ 6, 8, 10], [ 1, 3, 5], [ 6, 8, 10]]) >>> a[...,:3] array([[1, 2, 3], [6, 7, 8], [1, 2, 3], [6, 7, 8]]) >>> a[...,::3] array([[1, 4], [6, 9], [1, 4], [6, 9]])
a[...,2]:...表示遍历每行,2表示索引为2的所在列。
a[...,:2]:...表示遍历每行,:2表示索引为<2的0,1所在的列。
a[...,::2]:...表示遍历每行,2表示步长,选取多索引为0,2,4所在的列。
原文地址:https://www.cnblogs.com/liuxuanhe/p/9209090.html
时间: 2024-09-28 08:30:58