Numpy 仅保留数组中的几个元素（同时索引多个元素）_numpy 数组一次索引多个

作者：不正经 | 2024-04-04 04:03:22

踩

numpy 数组一次索引多个

我们有时候需要取一个数组中特定几个位置的元素，但是重新一个个索引出来再拼接会很麻烦，这种时候有一种比较神奇的索引方式，看代码！！！


import numpy as np
 
arr = np.array([1,2,3]) 
 
# 保留一个和第三个元素
arr = arr[[0,2]]

结果：[1 3]

希望我的分享能够帮到你~~~

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/不正经/article/detail/357264