时间戳

首页 > 解决方案 > 将浮点数随机四舍五入为整数

python - 将浮点数随机四舍五入为整数

问题描述

我想要一个函数（如果相关，使用 Python 3.6+），它将以下列方式将浮点数随机四舍五入为整数：

给定一个实数，x，让a = floor(x)和 让b = ceil(x)。然后，编写一个函数，该函数s_int()将以a的概率返回并以 的概率b - x返回。bx - a

例如，s_int(14.8)应该返回1420% 的时间，并15在剩余的 80% 的时间内返回。

这是我的尝试：

import math
from random import random

def s_int(x):
    a = math.floor(x)
    return a + ((x - a) > random())

它似乎适用于我能想到的所有情况：

In [2]: Counter(s_int(14.7) for _ in range(1000000))
Out[2]: Counter({14: 300510, 15: 699490})

In [3]: Counter(s_int(-14.7) for _ in range(1000000))
Out[3]: Counter({-15: 700133, -14: 299867})

In [4]: Counter(s_int(14) for _ in range(1000000))
Out[4]: Counter({14: 1000000})

In [5]: Counter(s_int(-14) for _ in range(1000000))
Out[5]: Counter({-14: 1000000})

In [6]: Counter(s_int(0) for _ in range(1000000))
Out[6]: Counter({0: 1000000})

以下是我的问题：

  1. 有没有我认为这个功能不起作用的边缘情况？

  2. 还有其他更简单或更优雅的解决方案吗？

  3. 可以让它运行得更快吗？

标签： pythonpython-3.xfloating-pointintegerrounding

解决方案

  1. 我认为没有边缘情况
  2. 尝试使用 numpy
import numpy as np
def s_int(x):
    a = np.floor(x)
    b = a + 1
    return (np.random.choice([a, b], p=[b - x, x - a]))
  1. 我认为这是一个 O(1) 操作。不会走得更快。

推荐阅读