adolfoalvarez / efficient_r_pazur Goto Github PK

using LinearAlgebra
using BenchmarkTools

function create_tridiagonal(n=5, d1=1, d2=2) 
	output_matrix = zeros(Int64, n, n)

	for row in 1:(n-1) 
    	output_matrix[row,row+1] = d2
    	if row > 1 
    		output_matrix[row,row-1] = d2
    	end

    end 

    row = n-1
  	output_matrix[row-1, row] = d2
  	output_matrix[row+1, row] = d2
  	output_matrix[diagind(output_matrix)] .= d1
  	
  	return(output_matrix)
end