Python diagonal示例

示例#1

 def test_diagonal_numpy_func(self):
     q = [[1, 2, 3], [1, 2, 3], [1, 2, 3]] * self.ureg.m
     self.assertQuantityEqual(np.diagonal(q, offset=-1),
                              [1, 2] * self.ureg.m)

示例#2

文件： test_numpy.py 项目： wiki01/pint

 def test_diagonal_numpy_func(self):
     q = [[1, 2, 3], [1, 2, 3], [1, 2, 3]] * self.ureg.m
     helpers.assert_quantity_equal(np.diagonal(q, offset=-1), [1, 2] * self.ureg.m)