numpy 中 digitize 函数的使用

青年有志

于 2023-03-24 10:24:20 发布

阅读量573

点赞数

分类专栏： # Python 文章标签： numpy python 开发语言

本文链接：https://blog.csdn.net/qq_46450354/article/details/129745386

版权

Python 专栏收录该内容

29 篇文章 7 订阅

订阅专栏

🏆 说明

numpy.digitize(x, bins, right = False)

参数：

x: numpy 数组
bins: 一维单调数组，必须是升序或者降序
right：间隔是否包含最右
return：x 在 bins 中的位置索引。

用法示例：

import random
import numpy as np
data = [random.randint(a=100, b = 10000) for _ in range(20)]
bins = [200, 300, 400, 500, 600, 700, 800, 900, 1000]
print('data: ', data)
print('bins: ', bins)
print('np.digitize(data, bins):', np.digitize(data, bins))

结果：
在这里插入图片描述

📋 解释：
digitize(data,bins) 函数执行的过程如下： 相当于两层 for 循环。首先最外面的一层：顺序的遍历 data 列表，取出当前的一个数 a，然后顺序的遍历 bins 列表，返回 bins 列表中某个数的索引，如何找呢？

规则如下： 如果 bins 列表是升序的，那么如果满足 bins[i-1] <= a < bins[i]，那么就保存 i，然后回到外层循环继续上面的操作，如果 bins 列表是降序的，那么如果满足 bins[i] <= a < bins[i-1], 那么就保存 i，当 data 列表遍历完之后，就返回由 i 组成的来源于 bins 的索引的一个索引列表。