ios 集合元素的筛选

这里的“集合”主要指NSArray，NSDictionary和NSSet以及其Mutable类型。

目录

• 1. 使用KVO
• 2. 使用NSPredicate
• 3. 使用集合本身的passingTest相关方法

 

返回目录

1. 使用KVO

可以使用KVO来选择一个集合中元素对应的属性。同时，KVO还有多种集合操作符可以实现对集合数据的快速分析，如下代码：

NSArray *array = @[@"mgen", @"tom", @"martin"];

//选择所有字符串的length为新的数组

NSLog(@"%@", [array valueForKeyPath:@"length"]);

//选择最大长度

NSLog(@"%@", [array valueForKeyPath:@"@max.length"]);

输出：

(

    4,

    3,

    6

)

6

 

还可以使用distinctUnionOfObjects操作符去取集合的不重复子集，这些操作符可以嵌套调用，比如，取一个数据的不重复子集，然后再输出总和，这里注意@distinctUnionOfObjects和@sum不能直接在一个keyPath中连接，所以需要两次调用valueForKeyPath：

NSArray *array = @[@1, @2, @2, @2, @2, @3];

//先取不重复的子集，然后计算总和

//注意@distinctUnionOfObjects和@sum不能直接在一个keyPath中连接，所以需要两次调用valueForKeyPath

NSLog(@"%@", [[array valueForKeyPath:@"@distinctUnionOfObjects.self"]valueForKeyPath:@"@sum.self"]);

输出：

6

因为集合的不重复子集就是1,2,3. 所以总和就是6.

 

返回目录

2. 使用NSPredicate

使用NSPredicate可以通过类似SQL的语句对集合进行筛选操作，方便快捷。

如下，找出长度大于3的字符串：

NSArray *array = @[@"mgen", @"tom", @"martin"];

//使用类似SQL语句的NSPredicate

NSPredicate *predicate = [NSPredicate predicateWithFormat:@"length > 3"];

NSLog(@"%@", [array filteredArrayUsingPredicate:predicate]);

输出：

(

    mgen,

    martin

)

 

因为是类SQL语言，所以NSPredicate自然支持多种方式的查询，比如下面，筛选出以m结尾或者长度大于4的字符串：

NSArray *array = @[@"mgen", @"tom", @"martin"];

//使用OR和SELF（数据成员本身）

NSPredicate *predicate = [NSPredicate predicateWithFormat:@"SELF ENDSWITH 'm' OR length > 4"];

NSLog(@"%@", [array filteredArrayUsingPredicate:predicate]);

输出：

(

    tom,

    martin

)

NSPredicate的LIKE操作符可以使用”?”和”*”这样的特殊字符来匹配字符串，同时MATCHES操作符还可以使用正则表达式，这里就不一一介绍了。

 

NSPredicate还有一个很无敌的创建方式，可以不适用类似SQL（以predicateWithFormat开头的方法）的语句，使用predicateWithBlock方法来通过一个Block来创建NSPredicate。

比如，把上面的NSPredicate用predicateWithBlock方法来实现：

NSArray *array = @[@"mgen", @"tom", @"martin"];

//万能的predicateWithBlock

NSPredicate *predicate = [NSPredicate predicateWithBlock:^BOOL(id evaluatedObject,NSDictionary *bindings) {

    NSString *str = evaluatedObject;

    return [str hasSuffix:@"m"] || str.length > 4;

}];

NSLog(@"%@", [array filteredArrayUsingPredicate:predicate]);

同样会输出正确的结果。

还可以使用NSCompoundPredicate和NSComparisonPredicate来创建NSPredicate，可以参考NSHipster。

 

 

返回目录

3. 使用集合本身的passingTest相关方法

集合本身通常会有passingTest相关方法，比如NSArray的的passingTest相关方法：

（截图来自官方文档）

 

passingTest相关方法通常是使用Block来筛选集合元素的，通常会返回NSIndexSet对象。可以使用objectsAtIndexes方法从NSIndexSet对象来返回子集合（关于NSIndexSet可以参考以前的一篇文章：iOS: 小议NSIndexSet）。passingTest相关方法有多种参数，比如可以设置集合遍历的方向等。同时passingTest相关方法的Block参数上会有集合元素的索引值（idx参数），并且可以命令停止立即停止遍历（stop参数）。

因此使用passingTest相关方法可以完成一些特殊的效果，比如从后往前筛选集合，返回第一个符合条件的成员。如下代码：

NSArray *array = @[@"mgen", @"tom", @"martin"];

//使用indexesOfObjectsWithOptions

//NSEnumerationReverse代表从后向前枚举

NSIndexSet *indexSet = [array indexesOfObjectsWithOptions:NSEnumerationReversepassingTest:^BOOL(id obj, NSUInteger idx, BOOL *stop)

{

    NSString *str = obj;

    if ([str hasSuffix:@"m"] || str.length > 4)

    {

        *stop = YES;

        return YES;

    }

    return NO;

}];

NSLog(@"%@", [array objectsAtIndexes:indexSet]);

输出：

(

    martin

)