Cody

Problem 1896. Index one element in each vector of an array along a given dimension

Solution 2084204

Submitted on 8 Jan 2020 by Tom Holz
This solution is locked. To view this solution, you need to provide a solution of the same size or smaller.
Solve This Problem View on Solution Map

Test Suite

Test Status Code Input and Output
1   Pass
for a = 1:30 ndms = ceil(rand(1) * 5) + 1 sz = 1 + ceil(rand(1, ndms) * 10); A = rand(sz); sz1 = sz(1); sz(1) = 1; I = ceil(rand(sz) * sz1); B = reshape(A(I(:)+(0:sz1:numel(A)-1)'), size(I)); p = randperm(ndms); A = permute(A, p); I = permute(I, p); B = permute(B, p); assert(isequal(dimsel(A, I), B)); end

ndms = 6 ndms = 6 ndms = 3 ndms = 6 ndms = 3 ndms = 3 ndms = 5 ndms = 6 ndms = 3 ndms = 2 ndms = 5 ndms = 5 ndms = 3 ndms = 2 ndms = 6 ndms = 5 ndms = 4 ndms = 6 ndms = 4 ndms = 6 ndms = 3 ndms = 5 ndms = 4 ndms = 5 ndms = 3 ndms = 3 ndms = 4 ndms = 2 ndms = 3 ndms = 6

Suggested Problems