NUMPY062 题解：上三角矩阵

题目

将矩阵转换为上三角矩阵。

解题思路

1. 读取矩阵
2. 使用 $\texttt{np.triu}$ 获取上三角部分

代码

import numpy as np

n = int(input())
arr = np.array([list(map(int, input().split())) for _ in range(n)])
result = np.triu(arr)
print(result)

代码详解

$\texttt{np.triu}$ 会将下三角部分设为 $0$，保留上三角部分。