How to define a function whose number of outputs depend on an input parameter value?

Question

Saurabh Madankar 2024-5-21

评论： Saurabh Madankar 2024-5-21

Say for M = 4, I have a code below.

function [c,ceq] = Constraint(M)
    c = [];
    for i=1:M
        ceq_{i} = i;
    end
    ceq = [ceq_{1} ceq_{2} ceq_{3} ceq_{4}];
end

My question is for bigger values of M, how do I define the last ceq so that I don't have to write each ceq_{i}'s individually?

Answer 1

Kunal Kandhari 2024-5-21

You can dynamically create and concatenate the elements of ceq based on the value of M. Here's how you can modify your function to achieve that:

function [c, ceq] = Constraint(M)

c = [];

ceq = [];

for i = 1:M

ceq = [ceq, i]; % Concatenate each element

end

With this modification, ceq will automatically adjust its size based on the value of M, eliminating the need to define each element individually.

Thanks a lot!