如何在python中选择n项并从ndarray跳过m？
提问于 2015-11-19 02:25:13
假设我有一个包含100个元素的ndarray，我想选择前4个元素，跳过6个元素然后继续这样做(换句话说，每10个元素选择前4个元素)。
我试过用step对python进行切片，但我认为它在我的情况下行不通。我怎么能这么做？我用的是潘达斯和朗皮，能帮上忙吗？我到处找，但没有发现像那样的切片。谢谢!
回答 5
Stack Overflow用户
回答已采纳
发布于 2015-11-19 03:08:25
你可以用NumPy slicing来解决你的案子。
对于1D数组的情况-
A.reshape(-1,10)[:,:4].reshape(-1)这可以扩展到2D数组的情况，选择沿着第一个轴-
A.reshape(-1,10,A.shape[1])[:,:4].reshape(-1,A.shape[1])Stack Overflow用户
发布于 2015-11-19 02:49:24
您可以将数组重组为一个10x10，然后使用切片来选择每一行的前4个元素。然后将经过整形、切片的数组压平：
In [46]: print a
[ 0 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24
25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49
50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74
75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99]
In [47]: print a.reshape((10,-1))[:,:4].flatten()
[ 0 1 2 3 10 11 12 13 20 21 22 23 30 31 32 33 40 41 42 43 50 51 52 53 60
61 62 63 70 71 72 73 80 81 82 83 90 91 92 93]Stack Overflow用户
发布于 2015-11-19 02:27:41
使用% 10
print [i for i in range(100) if i % 10 in (0, 1, 2, 3)]
[0, 1, 2, 3, 10, 11, 12, 13, 20, 21, 22, 23, 30, 31, 32, 33, 40, 41, 42, 43, 50, 51, 52, 53, 60, 61, 62, 63, 70, 71, 72, 73, 80, 81, 82, 83, 90, 91, 92, 93]页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/33801170
复制相关文章
腾讯云开发者
