numpy 1行流 import numpy as np

def make_diagonal(x): return np.eye(x.shape[0])*x

if name == "main": x = np.array(eval(input())) print(make_diagonal(x))