pymongo查询列表元素_pymongo的数组操作
数据样例
{"_id" : ObjectId("5e67343f00850012ec08ad3e"),"river": ["长江","黄河","淮河"]
}
{"_id" : ObjectId("5e67343f00850012ec08ad3f"),"river": ["长江","松花江","嘉陵江"]
}
一：普通包含查询
defget_all(self):
# 查询数组中包含长江的所有文档
result_cursor= self.db["test_info"].find({"river":"长江"})
temp=list()for result inresult_cursor:
temp.append(result)print(temp)returntemp
[{'_id': ObjectId('5e67343f00850012ec08ad3e'), 'river': ['长江', '黄河', '淮河']}, {'_id': ObjectId('5e67343f00850012ec08ad3f'), 'river': ['长江', '松花江', '嘉陵江']}]
二：$all 包含查询
数据样例
{"_id" : ObjectId("5e67343f00850012ec08ad3e"),"river": ["长江","黄河","淮河"]
}
{"_id" : ObjectId("5e67343f00850012ec08ad3f"),"river": ["长江","松花江","嘉陵江"]
}
{"_id" : ObjectId("5e6737552a89d7c2fc0cd5d1"),"river": ["长江","黄河","黄浦江"]
}
需要查询数组里面包含长江和黄河的所有文档
defget_all(self):
result_cursor= self.db["test_info"].find({"river":{"$all":["长江","黄河"]}})
temp=list()for result inresult_cursor:
temp.append(result)print(temp)returntemp
[{'_id': ObjectId('5e67343f00850012ec08ad3e'), 'river': ['长江', '黄河', '淮河']}, {'_id': ObjectId('5e6737552a89d7c2fc0cd5d1'), 'river': ['长江', '黄河', '黄浦江']}]
普通查询的 {"river":"长江"} 其实就相当于 {"river":{"$all":["长江"]}
三：指定位置(下标)查询 数组名称.index
要查询数组最后一位元素是黄浦江的文档
defget_all(self):
result_cursor= self.db["test_info"].find({"river.2":"黄浦江"})
temp=list()for result inresult_cursor:
temp.append(result)print(temp)returntemp
[{'_id': ObjectId('5e6737552a89d7c2fc0cd5d1'), 'river': ['长江', '黄河', '黄浦江']}]
四：指定长度查询 $size
查询数组长度为3的所有文档
defget_all_size(self):
result_cursor= self.db["test_info"].find({"river":{"$size":3}})
temp=list()for result inresult_cursor:
temp.append(result)print(temp)returntemp
[{'_id': ObjectId('5e67343f00850012ec08ad3e'), 'river': ['长江', '黄河', '淮河']}, {'_id': ObjectId('5e67343f00850012ec08ad3f'), 'river': ['长江', '松花江', '嘉陵江']}, {'_id': ObjectId('5e6737552a89d7c2fc0cd5d1'), 'river': ['长江', '黄河', '黄浦江']}]
五：返回数组的一个子集 $slice find的第二个查询参数
返回查询结果数组里面的前两个元素
defget_all_size(self):
result_cursor= self.db["test_info"].find({},{"river":{"$slice":2}})
temp=list()for result inresult_cursor:
temp.append(result)print(temp)returntemp
[{'_id': ObjectId('5e67343f00850012ec08ad3e'), 'river': ['长江', '黄河']}, {'_id': ObjectId('5e67343f00850012ec08ad3f'), 'river': ['长江', '松花江']}, {'_id': ObjectId('5e6737552a89d7c2fc0cd5d1'), 'river': ['长江', '黄河']}]
defget_all_size(self):
result_cursor= self.db["test_info"].find({},{"river":{"$slice":-1}})
temp=list()for result inresult_cursor:
temp.append(result)print(temp)returntemp
[ {'_id': ObjectId('5e67343f00850012ec08ad3e'), 'river': ['淮河']}, {'_id': ObjectId('5e67343f00850012ec08ad3f'), 'river': ['嘉陵江']}, {'_id': ObjectId('5e6737552a89d7c2fc0cd5d1'), 'river': ['黄浦江']}]
返回查询结果数组里面的最后一个元素
$slice可以接受偏移量，跳过指定数量的元素，返回指定数量的元素
defget_all_size(self):
result_cursor= self.db["test_info"].find({},{"river":{"$slice":[1,2]}})
temp=list()for result inresult_cursor:
temp.append(result)print(temp)returntemp
[ {'_id': ObjectId('5e67343f00850012ec08ad3e'), 'river': ['黄河', '淮河']}, {'_id': ObjectId('5e67343f00850012ec08ad3f'), 'river': ['松花江', '嘉陵江']}, {'_id': ObjectId('5e6737552a89d7c2fc0cd5d1'), 'river': ['黄河', '黄浦江']}]
跳过数组里面的第一个元素，返回跳过后的前两个元素：如果跳过第一个元素后，剩下的元素不够两个，就返回剩下的所有元素
数据格式如下
{"_id" : ObjectId("5e688d3a2a89d7c2fc0d64ee"),"t_id" : "1","data": [
{"time" : ISODate("2020-03-07T19:59:00.000Z"),"x" : "100"},
{"time" : ISODate("2020-03-08T19:59:00.000Z"),"x" : "200"},
{"time" : ISODate("2020-03-09T19:59:00.000Z"),"x" : "300"},
{"time" : ISODate("2020-03-10T19:59:00.000Z"),"x" : "400"},
{"time" : ISODate("2020-03-11T19:59:00.000Z"),"x" : "500"}
]
}
数组里面的元素，返回指定时间段内的数据，返回 2020-03-07-2020-03-11之间的data数据
defget_list_time_range(self):ifself.connect_result:
s_time= datetime.datetime(2020,3,7,0,0,0)
e_time= datetime.datetime(2020,3,11,23,59,59)
match_dict= {"$match":{"t_id":"1"}}
unwind_dict= {"$unwind":"$data"}
project_dict= {"$project":{"_id":0,"data":1}}
match_dict2= {"$match":{"data.time":{"$gte":s_time,"$lte":e_time}}}
result= self.db["test_info"].aggregate([match_dict,unwind_dict,project_dict,match_dict2])
temp_list=list()print(result)for info inresult:
info= info["data"]print(info)
temp_list.append((info["x"],100))print(temp_list)
结果
{'time': datetime.datetime(2020, 3, 7, 19, 59), 'x': '100'}
{'time': datetime.datetime(2020, 3, 8, 19, 59), 'x': '200'}
{'time': datetime.datetime(2020, 3, 9, 19, 59), 'x': '300'}
{'time': datetime.datetime(2020, 3, 10, 19, 59), 'x': '400'}
{'time': datetime.datetime(2020, 3, 11, 19, 59), 'x': '500'}
[('100', 100), ('200', 100), ('300', 100), ('400', 100), ('500', 100)]
六，数组插入数据
$push：将内容插入指定的数组里面，不论内容是否相同，不进行去重，均执行插入
#数据样式
{"_id" : ObjectId("5e8eba932a89d7c2fc29ebb3"),"name" : "a","data_list": []
}
需求一：给 data_list插入三组一样的数据
defpush_method(self):
result= self.db["test1"].update({"name":"a"},{"$push":{"data_list":{"city":"西安"}}})print(result)#结果
{'n': 1, 'nModified': 1, 'ok': 1.0, 'updatedExisting': True}
执行三次后的结果
{"_id" : ObjectId("5e8eba932a89d7c2fc29ebb3"),"name" : "a","data_list": [
{"city" : "西安"},
{"city" : "西安"},
{"city" : "西安"}
]
}
结果可以看出，相同的三组数据都可以插入，那么怎么让相同的数据，不能够插入呢？
#数据格式
{"_id" : ObjectId("5e8eba932a89d7c2fc29ebb3"),"name" : "a","data_list": []
}
$addToSet：将内容插入指定数组，内容相同就不执行插入，去重
执行三次的打印结果
defadd_to_set(self):
result= self.db["test1"].update({"name": "a"}, {"$addToSet": {"data_list": {"city": "西安"}}})print(result)#结果
{'n': 1, 'nModified': 1, 'ok': 1.0, 'updatedExisting': True}
{'n': 1, 'nModified': 0, 'ok': 1.0, 'updatedExisting': True}
{'n': 1, 'nModified': 0, 'ok': 1.0, 'updatedExisting': True}
执行三次的数据样式
/* 1 */{"_id" : ObjectId("5e8eba932a89d7c2fc29ebb3"),"name" : "a","data_list": [
{"city" : "西安"}
]
}
$pushAll：版本4.0.10已经作废，可使用 push代替，插入一组数据
#数据样式
{"_id" : ObjectId("5e8eba932a89d7c2fc29ebb3"),"name" : "a","data_list": []
}
MongoDB shell version v4.0.10
报错：
pymongo.errors.WriteError: Unknown modifier: $pushAll
defpush_many_method(self):
result= self.db["test1"].update({"name": "a"}, {"$push": {"data_list":[{"city": "西安"},{"city": "南京"}]}})print(result)#结果
{'n': 1, 'nModified': 1, 'ok': 1.0, 'updatedExisting': True}
/* 1 */{"_id" : ObjectId("5e8eba932a89d7c2fc29ebb3"),"name" : "a","data_list": [
[
{"city" : "西安"},
{"city" : "南京"}
]
]
}
可以看出，push也可以一次性，插入多组数据。
# TODO