Is there any way to save result of prediction as a boolean or number 1(True) or 0(False) in a matrix?

Question

I wonder about is there a method or sortcut to assign my prediction result as a number to a matrix? For example my matrix is result if model predicts first feature