Python-like comprehensions for NSArray

NSArray *test = @[@1, @2, @3, @4, @5];
NSArray *comp = [test arrayWithComprehension:^id(id obj) {
    return @([obj intValue] * [obj intValue]);
}];

Filtering is supported too.

NSArray *comp = [test arrayWithComprehension:^id(id obj) {
    return @([obj intValue] * [obj intValue]);
} ifCondition:^BOOL(id obj) {
    return ([obj intValue] % 2 == 0);
}];

Download on GitHub