根据python中的bin大小将数组划分为多个(单个)数组

时间:2012-07-01 03:11:26

标签: python

我正在阅读包含以下值的文件:

-0.68285 -6.919616
-0.7876 -14.521115
-0.64072 -43.428411
-0.05368 -11.561341
-0.43144 -34.768892
-0.23268 -10.793603
-0.22216 -50.341101
-0.41152 -90.083377
-0.01288 -84.265557
-0.3524 -24.253145

如何根据第1列中的值宽度为0.1?

将其拆分为单个数组

我希望我的输出像这样:

array1=[[-0.05368, -11.561341],[-0.01288, -84.265557]]
array2=[[-0.23268, -10.79360] ,[-0.22216, -50.341101]]
array3=[[-0.3524, -24.253145]]
array4=[[-0.43144, -34.768892], [-0.41152, -90.083377]]
array5=[[-0.68285, -6.919616],[-0.64072, -43.428411]]
array6=[[-0.7876, -14.521115]]

1 个答案:

答案 0 :(得分:0)

这是一个使用Python的圆函数和字典类的简单解决方案:

lines = open('some_file.txt').readlines()

dictionary = {}
for line in lines:
    nums = line[:-1].split(' ')     #remove the newline and split the columns
    k = round(float(nums[0]), 1)    #round the first column to get the bucket
    if k not in dictionary:         
        dictionary[k] = []          #add an empty bucket
    dictionary[k].append([float(nums[0]), float(nums[1])])
                                    #add the numbers to the bucket
print dictionary    

要获得特定的存储桶(如.3),只需执行以下操作:

x = dictionary[0.3]

x = dictionary.get(0.3, [])

如果您只想为空桶返回一个空列表。