Python where_indicesの例

コード例 #1

ファイル: rubik_test_comparison.py プロジェクト: hiive/rubik

 def where_indices_condition(self):
     a = np.eye(3)
     a[1, 2] = -2.0
     a[2, 1] = 2.0
     cube = cb.where_indices(a, np.abs(a) > 1.0)
     self.assertEqual(cube.shape, (2, 3))
     self.assertCubesAreEqual(cube[0], np.array([1.0, 2.0, -2.0]))
     self.assertCubesAreEqual(cube[1], np.array([2.0, 1.0, 2.0]))

コード例 #2

ファイル: rubik_test_comparison.py プロジェクト: Borsos/rubik

 def where_indices_condition(self):
     a = np.eye(3)
     a[1, 2] = -2.0
     a[2, 1] = 2.0
     cube = cb.where_indices(a, np.abs(a) > 1.0)
     self.assertEqual(cube.shape, (2, 3))
     self.assertCubesAreEqual(cube[0], np.array([1.0, 2.0, -2.0]))
     self.assertCubesAreEqual(cube[1], np.array([2.0, 1.0, 2.0]))

コード例 #3

ファイル: rubik_test_comparison.py プロジェクト: hiive/rubik

 def where_indices(self):
     a = np.eye(3)
     a[1, 2] = -2.0
     a[2, 1] = 2.0
     cube = cb.where_indices(a)
     self.assertEqual(cube.shape, (5, 3))
     self.assertCubesAreEqual(cube[0], np.array([0.0, 0.0, 1.0]))
     self.assertCubesAreEqual(cube[1], np.array([1.0, 1.0, 1.0]))
     self.assertCubesAreEqual(cube[2], np.array([1.0, 2.0, -2.0]))
     self.assertCubesAreEqual(cube[3], np.array([2.0, 1.0, 2.0]))
     self.assertCubesAreEqual(cube[4], np.array([2.0, 2.0, 1.0]))

コード例 #4

ファイル: rubik_test_comparison.py プロジェクト: Borsos/rubik

 def where_indices(self):
     a = np.eye(3)
     a[1, 2] = -2.0
     a[2, 1] = 2.0
     cube = cb.where_indices(a)
     self.assertEqual(cube.shape, (5, 3))
     self.assertCubesAreEqual(cube[0], np.array([0.0, 0.0, 1.0]))
     self.assertCubesAreEqual(cube[1], np.array([1.0, 1.0, 1.0]))
     self.assertCubesAreEqual(cube[2], np.array([1.0, 2.0, -2.0]))
     self.assertCubesAreEqual(cube[3], np.array([2.0, 1.0, 2.0]))
     self.assertCubesAreEqual(cube[4], np.array([2.0, 2.0, 1.0]))