东方耀AI技术分享

标题: 理解np.partition与np.argpartition [打印本页]

---

作者: 东方耀    时间: 2019-11-30 11:57
标题: 理解np.partition与np.argpartition
理解np.partition与np.argpartition

1. 
   
2. x = np.random.permutation(np.arange(10))
   
3. 
   
4. print(x)
   
5. 
   
6. for i in range(10):
   
7.     # kth : 它不是指序列的第k个元素，而是指序列的第k大
   
8.     # np.partition()实现 无序数组第k大数字的问题
   
9.     print('k=%d' % i, np.partition(x, kth=i))
   
10.     print('相对原始序列的索引：', np.argpartition(x, kth=i))
    
11.     print('第%d大的值：' % i, np.partition(x, kth=i)[i])
    
12. 
    
13. print(sorted(x))

复制代码

---

欢迎光临 东方耀AI技术分享 (http://www.ai111.vip/)

Powered by Discuz! X3.4