【发布时间】:2020-04-17 23:47:24
【问题描述】:
所以,我有一个 numpy 数组,我想计算元素在特定间隔内出现的频率。 例如,
array = np.array([0, 1, 1, 1, 2, 3, 4, 5])
intervals = np.array([0., 0.5, 1., 1.5, 2., 2.5, 3., 3.5, 4., 4.5, 5.])
result = {0.5: 0.125, 1.5: 0.375, 2.5: 0.125, 3.5: 0.125, 4.5: 0.125}
我的代码运行良好,但对我来说看起来很乱
import numpy as np
from collections import Counter
def freqs(arr):
#defining our intervals
intervals = np.arange(round(np.min(arr)), round(np.max(arr))+0.5, 0.5)
frequency = list()
#going through every number in array, if smaller then interval's value, appending interval's value
for arr_i in arr:
for intr_j in intervals:
if arr_i < intr_j:
frequency.append(intr_j)
break
#counting intervals' values
dic = dict(Counter(frequency))
#divide dic's values by lenghth of an array
freqs = dict(zip(list(dic.keys()), (np.array(list(dic.values())))/len(arr)))
return freqs
我不喜欢的部分是我们将字典的值除以数组的长度,并使用大量的构造来声明新字典。但是我们所做的一切只是将值除以某个数字。
【问题讨论】:
-
感谢大家的帮助!
标签: python arrays numpy intervals frequency