2017-02-20 2 views
0

Я хотел бы знать, как я могу выбрать значения с более чем одним повторением при использовании случайного числа, чтобы случайным образом генерировать числа. я делаю следующее:Выберите значения с более чем одним повторением при использовании случайного numpy?

1) Сначала приготовьте число между 1-10

ran = ceil(np.random.random(10)*10) 
print ran 
[ 2. 9. 8. 9. 10. 8. 10. 7. 1. 1.] 

2) Теперь, используя itemfreq от scipy.stats, чтобы получить таблицу частот , где второй столбец частота ,

freq_tmp = itemfreq(ran) 
print freq_tmp 
[[ 1. 2.] 
[ 2. 1.] 
[ 7. 1.] 
[ 8. 2.] 
[ 9. 2.] 
[ 10. 2.]] 

print freq_tmp[0:][:,1] 

[ 2. 1. 1. 2. 2. 2.] 

это показать мне только частоту, но я хотел бы знать значения с более чем 1 повторение, в этом примере, должны быть:

[1. 8. 9. 10] 

Спасибо за вашу помощь!

+0

'' 'freq_tmp [:, 0] [freq_tmp [ :, 1]> 1] '' ' – wwii

ответ

1

Вы почти находитесь. Вы можете просто поставить > в ломтерезки так:

import numpy as np 

a = np.array([[1,2],[2,1],[7,1],[8,2],[9,2],[10,2]]) 

print(a[a[:,1] > 1][:,0]) 

выход:

[ 1 8 9 10] 
1

Другой NumPy решение:

In [57]: ran=randint(1,11,10) 

In [58]: ran 
Out[58]: array([3, 4, 6, 1, 9, 4, 2, 8, 6, 8]) 

In [59]: uniqs,cnts=np.unique(ran,return_counts=True) 

In [60]: uniqs,cnts 
Out[60]: (array([1, 2, 3, 4, 6, 8, 9]), array([1, 1, 1, 2, 2, 2, 1], dtype=int64)) 

In [61]: uniqs[cnts>=2] 
Out[61]: array([4, 6, 8]) 
Смежные вопросы