numpy.tril #
- 麻木的。tril ( m , k = 0 ) [来源] #
数组的下三角形。
返回一个数组的副本,其中第 k个对角线上方的元素已为零。对于
ndim
超过 2 的数组,tril
将应用于最后两个轴。- 参数:
- m类似数组,形状 (…, M, N)
输入数组。
- k整数,可选
其上方的对角线为零元素。 k = 0(默认值)为主对角线,k < 0位于其下方,k > 0位于其上方。
- 返回:
- tril ndarray,形状 (…, M, N)
m的下三角形,其形状和数据类型与m相同。
也可以看看
triu
同样的事情,仅适用于上三角形
例子
>>> 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]]])