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)