id: search_vector_python.md

查询向量

Milvus 支持在集合或分区中查询向量。

在集合中查询向量

1. 创建搜索参数。搜索参数是一个 JSON 字符串，在 Python SDK 中以字典来表示。

   >>> search_param = {'nprobe': 16}

   对于不同的索引类型，搜索所需参数也有区别。所有的搜索参数都必须赋值。详细信息请参考 Milvus 索引类型。

2. 创建随机向量作为 query_records 进行搜索：

   # Create 5 vectors of 256 dimensions.
   >>> q_records = [[random.random() for _ in range(256)] for _ in range(5)]
   >>> milvus.search(collection_name='test01', query_records=q_records, top_k=2, params=search_param)

   • top_k 指的是向量空间中距离目标向量最近的 k 个向量。
   • top_k 的范围为：[1, 16384]。

在分区中查询向量

# Create 5 vectors of 256 dimensions.
>>> q_records = [[random.random() for _ in range(256)] for _ in range(5)]
>>> milvus.search(collection_name='test01', query_records=q_records, top_k=1, partition_tags=['tag01'], params=search_param)

常见问题