博客
关于我
python numpy矩阵索引_python – 在2D numpy ndarray或numpy矩阵中获取前N个值的索引
阅读量:803 次
发布时间:2023-03-06

本文共 832 字,大约阅读时间需要 2 分钟。

为了找到给定N维向量数据中各向量成对之间的最大距离及其对应的索引,可以按照以下步骤进行:

  • 计算距离矩阵:使用sklearn的pairwise_distances函数生成所有向量之间的距离矩阵。

  • 生成索引对:生成所有i < j的索引对,提取这些对应的距离值。

  • 找到最大距离:在这些对应的距离中找到最大值。

  • 确定索引对:记录达到最大距离的所有索引对,并确保i < j。

  • 以下是具体的代码实现:

    import numpy as npfrom sklearn.metrics.pairwise import pairwise_distances# 假设数据已经定义为datadata = np.array([[5, 6, 1], [2, 0, 8], [4, 9, 3]])# 计算距离矩阵dists = pairwise_distances(data)# 生成所有i < j的索引对i, j = np.ndindex(len(data), len(data))mask = i < jdistances = dists[i, j]# 找到最大距离max_distance = np.max(distances)# 找到所有达到最大距离的索引对max_indices = np.where(distances == max_distance)unique_pairs = set(zip(max_indices[0], max_indices[1]))# 确保i < j并返回结果for pair in unique_pairs:    i, j = pair    if i < j:        print(f"最大距离为{max_distance},对应的索引为{i}和{j}")        break

    输出结果

    最大距离为10.48808848,对应的索引为1和2。

    通过上述方法,可以自动识别出数据中各向量成对之间的最大距离及其索引,适用于N维向量数据。

    转载地址:http://goafk.baihongyu.com/

    你可能感兴趣的文章
    Python You are using pip version 10.0.1, however version 19.3.1 is available.
    查看>>
    python zipfile模块学习笔记(一)
    查看>>
    Python zip函数 详解(全)
    查看>>
    Python \r\n与\n的转换
    查看>>
    python __new__中单例的作用
    查看>>
    python | aiofiles,一个超酷的 Python 库!
    查看>>
    python | akshare,一个超强的 开源Python 金融数据接口库!
    查看>>
    python | alabaster,一个强大的 关于alabaster 主题 Python 库!
    查看>>
    python | algorithms,一个超赞的 集合常用算法的Python 库!
    查看>>
    python调用jpype 报错:OSError JVM is already started和JVM cannot be restarted
    查看>>
    python | authlib,一个强大的 Python 库!
    查看>>
    python | awswrangler,一个高效的 Python 库!
    查看>>
    python | bashplotlib,一个有趣的Python库!
    查看>>
    python | bentoml,一个超级厉害的 模型部署 Python 库!
    查看>>
    python调用jar包的模块_python调用jar包
    查看>>
    python | black,一个神奇的 代码格式化工具 Python 库!
    查看>>
    python | bleach,一个超强的 Python 库!
    查看>>
    python | cartopy,一个有趣的 Python 库!
    查看>>
    python | cloud-init,一个实用的 云计算 Python 库!
    查看>>
    python | code2flow,一个神奇的 Python 库!
    查看>>