示例

1. 安装 PyMongo

在 Python 环境中安装 PyMongo，可以使用 pip install pymongo 命令进行安装。

2. 连接 MongoDB 数据库

使用 PyMongo 连接 MongoDB 数据库，需要指定数据库的地址、端口号、用户名和密码等参数。

import pymongo
from pymongo import MongoClient

client = MongoClient('mongodb://username:password@localhost:27017/')
db = client['mydatabase']

3. 创建地理空间索引

在 MongoDB 中，需要创建地理空间索引才能执行地理空间查询操作。在 PyMongo 中，可以使用 create_index() 方法创建地理空间索引。

db.places.create_index([("location", "2dsphere")])

4. 插入地理位置数据

在 MongoDB 中，可以使用 GeoJSON 格式保存地理位置数据。在 PyMongo 中，可以使用 insert_one 或者 insert_many 方法进行插入操作。

data = {
"name": "pidancode.com",
"location": {
"type": "Point",
"coordinates": [10.001, 30.002]
}
}

result = db.places.insert_one(data)

5. 执行地理空间查询操作

在 MongoDB 中，可以使用 $near、$geoWithin、$geoIntersects 等操作符执行地理空间查询操作。在 PyMongo 中，可以使用 find() 方法和 $near、$geoWithin、$geoIntersects 等操作符进行查询操作。

查询最近的两个地理位置

cursor = db.places.find(
{
"location":{
"$near":{
"$geometry":{
"type":"Point",
"coordinates":[10.000, 30.000]
},
"$maxDistance":100000
}
}
}
).limit(2)

for document in cursor:
print(document)

查询在一个多边形区域内的地理位置

cursor = db.places.find(
{
"location":{
"$geoWithin":{
"$geometry":{
"type": "Polygon",
"coordinates": [[
[10.000, 30.000],
[11.000, 30.000],
[11.000, 31.000],
[10.000, 31.000],
[10.000, 30.000]
]]
}
}
}
}
)

for document in cursor:
print(document)

查询与一个地理位置相交的地理位置

cursor = db.places.find(
{
"location":{
"$geoIntersects":{
"$geometry":{
"type":"Point",
"coordinates":[10.001, 30.002]
}
}
}
}
)

for document in cursor:
print(document)

6. 关闭连接

在使用完 PyMongo 连接 MongoDB 数据库后，需要使用 close() 方法关闭连接。

client.close()