题解 | 将向量转换为对角矩阵

将向量转换为对角矩阵

https://www.nowcoder.com/practice/fa80e7e3e07345b1993b93f46182643b

import numpy as np

def make_diagonal(x):
    arr = np.array( x , dtype=np.float32 )
    return np.diag(arr)
    
    
if __name__ == "__main__":
    x = np.array(eval(input()))
    print(make_diagonal(x))


全部评论

相关推荐

评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务