python 从数组中随机选择一部分取出,并从原数组中删除取出的内容

可以使用numpy中的random.choice函数随机选择数组的下标

numpy.random.choice(a, size=None, replace=True, p=None)
#a为一维数组或int值,为一维数组时会从该数组中随机选择元素,为int值时会先生成一个np.arange(a)的数组,然后从中随机选择元素
#sizeint值,为选择元素的个数
#replace默认为True,意思是随机选择出的元素还会放回样本集中,即生成的数列中可能存在相同的元素,为False时就不会出现
#p为样本选择的概率,默认为一致分布
import numpy as np
data=np.random.random(size=10)
data
Out[4]: 
array([ 0.21575642,  0.30620622,  0.01454852,  0.46253994,  0.11222712,
        0.32893411,  0.11040516,  0.51010326,  0.83162364,  0.84285834])
index_1=np.random.choice(data.shape[0],4,replace=False)
index_1
Out[6]: array([1, 4, 2, 3])
data1=data[index_1]
data1
Out[8]: array([ 0.30620622,  0.11222712,  0.01454852,  0.46253994])

然后我们如何获得剩下的数组中的内容呢,我们可以先生成原数组所有的下标,然后用np.delete函数删除之前随机生成的下标数组

numpy.delete(arr,obj,axis=None) 
#arr:输入向量 
#obj:表明哪一个子向量应该被移除。可以为整数或一个int型的向量 
#axis:表明删除哪个轴的子向量,若默认,则返回一个被拉平的向量
index_2=np.arange(data.shape[0])
index_2
Out[10]: array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])
index_2=np.delete(index_2,index_1)
index_2
Out[12]: array([0, 5, 6, 7, 8, 9])
data2=data[index_2]
data2
Out[14]: 
array([ 0.21575642,  0.32893411,  0.11040516,  0.51010326,  0.83162364,
        0.84285834])
  • 18
    点赞
  • 48
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值