时间戳

首页 > 解决方案 > 如何将字符串转换为数字范围（浮点数）？

python - 如何将字符串转换为数字范围（浮点数）？

问题描述

我有一排字符串：

poor: <=0.75; intermediate: >0.75 & <=1.25; normal: >1.25 & <= 2.5; High inducibility: >2.5

我想根据上面的字符串创建一个字典，如下所示：

import numpy
dict = {poor: numpy.arange(0, 0.75, 0.05),
        intermediate: numpy.arange(0.75, 1.25, 0.05),
        normal: numpy.arange(1.25, 2.5, 0.05),
        High: numpy.arange(2.5, 5, 0.05)}

那么，在 python 2.7 中有什么好的方法吗？

谢谢！

标签： pythonpython-2.7

解决方案

使用np.digitize：

import numpy as np
from pprint import pprint

bins = [0.75, 1.25, 2.5]
bin_names = np.array(['poor', 'intermediate', 'normal', 'high'])

data = np.linspace(0, 4, 10, endpoint=False)
result = bin_names[np.digitize(data, bins)]

pprint(list(zip(np.round(data, 1), result)))

输出：

[(0.0, 'poor'),
 (0.4, 'poor'),
 (0.8, 'intermediate'),
 (1.2, 'intermediate'),
 (1.6, 'normal'),
 (2.0, 'normal'),
 (2.4, 'normal'),
 (2.8, 'high'),
 (3.2, 'high'),
 (3.6, 'high')]

推荐阅读