numpy.tril#
- numpy.tril(m, k=0)[source]#
数组的下三角。
返回一个数组的副本,其中高于第 k 条对角线的元素归零。对于
ndim
超过 2 的数组,tril
将应用于最后两个轴。- 参数:
- marray_like,形状 (…, M, N)
输入数组。
- kint,可选
要将元素归零的对角线上方。 k = 0(默认值)为主对角线, k < 0 在其下方, k > 0 在其上方。
- 返回值:
- trilndarray,形状 (…, M, N)
m 的下三角,与 m 形状和数据类型相同。
另请参阅
triu
相同的功能,但仅适用于上三角
示例
>>> import numpy as np >>> np.tril([[1,2,3],[4,5,6],[7,8,9],[10,11,12]], -1) array([[ 0, 0, 0], [ 4, 0, 0], [ 7, 8, 0], [10, 11, 12]])
>>> np.tril(np.arange(3*4*5).reshape(3, 4, 5)) array([[[ 0, 0, 0, 0, 0], [ 5, 6, 0, 0, 0], [10, 11, 12, 0, 0], [15, 16, 17, 18, 0]], [[20, 0, 0, 0, 0], [25, 26, 0, 0, 0], [30, 31, 32, 0, 0], [35, 36, 37, 38, 0]], [[40, 0, 0, 0, 0], [45, 46, 0, 0, 0], [50, 51, 52, 0, 0], [55, 56, 57, 58, 0]]])