Как bin серии значений float в гистограмму в Python?
у меня есть набор значений в float (всегда меньше 0). Что я хочу bin в гистограмме, я,Эл. каждый бар гистограммы содержать диапазон значений [0,0.150)
данные, которые у меня есть, выглядят так:
0.000
0.005
0.124
0.000
0.004
0.000
0.111
0.112
С мой код ниже, я ожидаю получить результат, который выглядит как
[0, 0.005) 5
[0.005, 0.011) 0
...etc..
Я попытался сделать такой биннинг с этим моим кодом. Но это, кажется, не работает. Как правильно это сделать?
#! /usr/bin/env python
import fileinput, math
log2 = math.log(2)
def getBin(x):
return int(math.log(x+1)/log2)
diffCounts = [0] * 5
for line in fileinput.input():
words = line.split()
diff = float(words[0]) * 1000;
diffCounts[ str(getBin(diff)) ] += 1
maxdiff = [i for i, c in enumerate(diffCounts) if c > 0][-1]
print maxdiff
maxBin = max(maxdiff)
for i in range(maxBin+1):
lo = 2**i - 1
hi = 2**(i+1) - 1
binStr = '[' + str(lo) + ',' + str(hi) + ')'
print binStr + 't' + 't'.join(map(str, (diffCounts[i])))
~
3 ответов
когда это возможно, не изобретайте колесо. NumPy имеет все, что вам нужно:
#!/usr/bin/env python
import numpy as np
a = np.fromfile(open('file', 'r'), sep='\n')
# [ 0. 0.005 0.124 0. 0.004 0. 0.111 0.112]
# You can set arbitrary bin edges:
bins = [0, 0.150]
hist, bin_edges = np.histogram(a, bins=bins)
# hist: [8]
# bin_edges: [ 0. 0.15]
# Or, if bin is an integer, you can set the number of bins:
bins = 4
hist, bin_edges = np.histogram(a, bins=bins)
# hist: [5 0 0 3]
# bin_edges: [ 0. 0.031 0.062 0.093 0.124]
from pylab import *
data = []
inf = open('pulse_data.txt')
for line in inf:
data.append(float(line))
inf.close()
#binning
B = 50
minv = min(data)
maxv = max(data)
bincounts = []
for i in range(B+1):
bincounts.append(0)
for d in data:
b = int((d - minv) / (maxv - minv) * B)
bincounts[b] += 1
# plot histogram
plot(bincounts,'o')
show()
первая ошибка-это:
Traceback (most recent call last):
File "C:\foo\foo.py", line 17, in <module>
diffCounts[ str(getBin(diff)) ] += 1
TypeError: list indices must be integers
почему вы преобразуете int в str, когда str необходим? Исправьте это, тогда мы получим:
Traceback (most recent call last):
File "C:\foo\foo.py", line 17, in <module>
diffCounts[ getBin(diff) ] += 1
IndexError: list index out of range
потому что вы сделали только 5 ведер. Я не понимаю вашего группируете схемы, но давайте сделаем это 50 ведер и посмотреть, что происходит:
6
Traceback (most recent call last):
File "C:\foo\foo.py", line 21, in <module>
maxBin = max(maxdiff)
TypeError: 'int' object is not iterable
maxdiff
является единственным значением из вашего списка ints, так что это max
здесь делаешь? Удалите его, теперь мы получаем:
6
Traceback (most recent call last):
File "C:\foo\foo.py", line 28, in <module>
print binStr + '\t' + '\t'.join(map(str, (diffCounts[i])))
TypeError: argument 2 to map() must support iteration
конечно, вы используете одно значение в качестве второго аргумента map
. Давайте упростим последние две строки из этого:
binStr = '[' + str(lo) + ',' + str(hi) + ')'
print binStr + '\t' + '\t'.join(map(str, (diffCounts[i])))
для этого:
print "[%f, %f)\t%r" % (lo, hi, diffCounts[i])
теперь он печатает:
6
[0.000000, 1.000000) 3
[1.000000, 3.000000) 0
[3.000000, 7.000000) 2
[7.000000, 15.000000) 0
[15.000000, 31.000000) 0
[31.000000, 63.000000) 0
[63.000000, 127.000000) 3
Я не уверен, что еще здесь делать, так как я действительно не понимаю, что вы надеетесь использовать. Кажется, это связано с двоичными силами, но для меня это не имеет смысла...