pymongo的数组操作

数据样例

{
    "_id" : ObjectId("5e67343f00850012ec08ad3e"),
    "river" : [ 
        "长江", 
        "黄河", 
        "淮河"
    ]
}


{
    "_id" : ObjectId("5e67343f00850012ec08ad3f"),
    "river" : [ 
        "长江", 
        "松花江", 
        "嘉陵江"
    ]
}

一:普通包含查询

    def get_all(self):
     # 查询数组中包含长江的所有文档 result_cursor
= self.db["test_info"].find({"river":"长江"}) temp = list() for result in result_cursor: temp.append(result) print(temp) return temp [{'_id': ObjectId('5e67343f00850012ec08ad3e'), 'river': ['长江', '黄河', '淮河']}, {'_id': ObjectId('5e67343f00850012ec08ad3f'), 'river': ['长江', '松花江', '嘉陵江']}]

二:$all 包含查询

数据样例

{
    "_id" : ObjectId("5e67343f00850012ec08ad3e"),
    "river" : [ 
        "长江", 
        "黄河", 
        "淮河"
    ]
}


{
    "_id" : ObjectId("5e67343f00850012ec08ad3f"),
    "river" : [ 
        "长江", 
        "松花江", 
        "嘉陵江"
    ]
}

{
    "_id" : ObjectId("5e6737552a89d7c2fc0cd5d1"),
    "river" : [ 
        "长江", 
        "黄河", 
        "黄浦江"
    ]
}

需要查询数组里面包含长江和黄河的所有文档

    def get_all(self):
        result_cursor = self.db["test_info"].find({"river":{"$all":["长江","黄河"]}})
        temp = list()
        for result in result_cursor:
            temp.append(result)
        print(temp)
        return temp


[{'_id': ObjectId('5e67343f00850012ec08ad3e'), 'river': ['长江', '黄河', '淮河']}, {'_id': ObjectId('5e6737552a89d7c2fc0cd5d1'), 'river': ['长江', '黄河', '黄浦江']}]

普通查询的 {"river":"长江"} 其实就相当于 {"river":{"$all":["长江"]}

三:指定位置(下标)查询  数组名称.index

要查询数组最后一位元素是黄浦江的文档

    def get_all(self):
        result_cursor = self.db["test_info"].find({"river.2":"黄浦江"})
        temp = list()
        for result in result_cursor:
            temp.append(result)
        print(temp)
        return temp


[{'_id': ObjectId('5e6737552a89d7c2fc0cd5d1'), 'river': ['长江', '黄河', '黄浦江']}]

四:指定长度查询 $size

查询数组长度为3的所有文档

    def get_all_size(self):
        result_cursor = self.db["test_info"].find({"river":{"$size":3}})
        temp = list()
        for result in result_cursor:
            temp.append(result)
        print(temp)
        return temp


[{'_id': ObjectId('5e67343f00850012ec08ad3e'), 'river': ['长江', '黄河', '淮河']}, {'_id': ObjectId('5e67343f00850012ec08ad3f'), 'river': ['长江', '松花江', '嘉陵江']}, {'_id': ObjectId('5e6737552a89d7c2fc0cd5d1'), 'river': ['长江', '黄河', '黄浦江']}]

五:返回数组的一个子集 $slice      find的第二个查询参数

返回查询结果数组里面的前两个元素

    def get_all_size(self):
        result_cursor = self.db["test_info"].find({},{"river":{"$slice":2}})
        temp = list()
        for result in result_cursor:
            temp.append(result)
        print(temp)
        return temp


[{'_id': ObjectId('5e67343f00850012ec08ad3e'), 'river': ['长江', '黄河']}, {'_id': ObjectId('5e67343f00850012ec08ad3f'), 'river': ['长江', '松花江']}, {'_id': ObjectId('5e6737552a89d7c2fc0cd5d1'), 'river': ['长江', '黄河']}]
    def get_all_size(self):
        result_cursor = self.db["test_info"].find({},{"river":{"$slice":-1}})
        temp = list()
        for result in result_cursor:
            temp.append(result)
        print(temp)
        return temp


[ {'_id': ObjectId('5e67343f00850012ec08ad3e'), 'river': ['淮河']}, {'_id': ObjectId('5e67343f00850012ec08ad3f'), 'river': ['嘉陵江']}, {'_id': ObjectId('5e6737552a89d7c2fc0cd5d1'), 'river': ['黄浦江']}]

返回查询结果数组里面的最后一个元素

$slice可以接受偏移量,跳过指定数量的元素,返回指定数量的元素

    def get_all_size(self):
        result_cursor = self.db["test_info"].find({},{"river":{"$slice":[1,2]}})
        temp = list()
        for result in result_cursor:
            temp.append(result)
        print(temp)
        return temp


[ {'_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数据

    def get_list_time_range(self):
        if self.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 in result:
                info = info["data"]
                print(info)
                temp_list.append((info["x"],100))
            print(temp_list)

结果

<pymongo.command_cursor.CommandCursor object at 0x0000000002FF3D88>
{'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插入三组一样的数据

def push_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:将内容插入指定数组,内容相同就不执行插入,去重

执行三次的打印结果

def add_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

def push_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

原文地址:https://www.cnblogs.com/meloncodezhang/p/12456032.html